Spaces:
Running
on
Zero
Running
on
Zero
change classifier model
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ hf_api_key = os.getenv("API_KEY")
|
|
17 |
login(token=hf_api_key)
|
18 |
|
19 |
classifier = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
|
20 |
-
generator = pipeline("
|
21 |
|
22 |
# Function to classify customer comments
|
23 |
def classify_comments():
|
|
|
17 |
login(token=hf_api_key)
|
18 |
|
19 |
classifier = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
|
20 |
+
generator = pipeline("text2text-generation", model="google/flan-t5-base")
|
21 |
|
22 |
# Function to classify customer comments
|
23 |
def classify_comments():
|