EnzoBustos commited on
Commit
f0fae67
1 Parent(s): 9e6d4d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -87,10 +87,10 @@ def theme_classification(text, text_classifier):
87
 
88
  return str(results["labels"][index])
89
 
90
- m2m100 = pipeline("translation_pt_to_en", src_lang="pt", tgt_lang="en", model="facebook/m2m100_418M")
91
- opus = pipeline("translation_pt_to_en", src_lang="pt", tgt_lang="en", model="Helsinki-NLP/opus-mt-mul-en")
92
- finbert = pipeline("sentiment-analysis",model="ProsusAI/finbert", tokenizer="ProsusAI/finbert")
93
- classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli",tokenizer="joeddav/xlm-roberta-large-xnli")
94
  sid_obj = SentimentIntensityAnalyzer()
95
 
96
 
 
87
 
88
  return str(results["labels"][index])
89
 
90
+ m2m100 = pipeline("translation_pt_to_en", src_lang="pt", tgt_lang="en", model="models/m2m100")
91
+ opus = pipeline("translation_pt_to_en", src_lang="pt", tgt_lang="en", model="models/opus")
92
+ finbert = pipeline("sentiment-analysis",model="models/finbert")
93
+ classifier = pipeline("zero-shot-classification", model="models/roberta")
94
  sid_obj = SentimentIntensityAnalyzer()
95
 
96