ZennyKenny commited on
Commit
3ae1eb6
Β·
verified Β·
1 Parent(s): 3fac692

change classifier model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("text-generation", model="mrm8488/t5-base-finetuned-question-generation-ap")
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():