Update app.py

app.py

@@ -23,11 +23,9 @@ def summarize_text(text, model_path="leadingbridge/summarization"):
     return summary
 
 # Sentiment Analysis Pre-Trained Model
-model_path = "leadingbridge/sentiment-analysis"
-tokenizer = BertTokenizerFast.from_pretrained(model_path)
-model = TFBertForSequenceClassification.from_pretrained(model_path, id2label={0: 'negative', 1: 'positive'} )
-
-def sentiment_analysis(text):
+def sentiment_analysis(text, model_path="leadingbridge/sentiment-analysis", id2label={0: 'negative', 1: 'positive'}):
+  tokenizer = BertTokenizerFast.from_pretrained(model_path)
+  model = TFBertForSequenceClassification.from_pretrained(model_path, id2label=id2label)
   pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
   result = pipe(text)
   return result