acecalisto3 commited on
Commit
95ec385
1 Parent(s): 7d4516a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ class App(gr.Blocks): # Corrected class inheritance
62
  "google/flan-t5-small",
63
  "facebook/bart-large-cnn", # Summarization
64
  "gpt2", # Text Generation
65
- "distilbert-base-uncased-finetuned-sst-2-english" # Sentiment Analysis
66
  # ... Other NLP model names from Hugging Face
67
  ]
68
  self.nlp_models = []
 
62
  "google/flan-t5-small",
63
  "facebook/bart-large-cnn", # Summarization
64
  "gpt2", # Text Generation
65
+ "CodeGenForCausalLM" # Sentiment Analysis
66
  # ... Other NLP model names from Hugging Face
67
  ]
68
  self.nlp_models = []