Update app.py

app.py

@@ -4,13 +4,13 @@ from transformers import pipeline
 # Summarization
 def summarization(text):
  text_model = pipeline("text-generation", model="ainize/bart-base-cnn")
- summary = text_model(text, max_length=100, do_sample=False)[0]["generated_text"]
+ summary = text_model(text, max_length=100, temperature=1.0)[0]["generated_text"]
  return summary
 
 # Sentiment Classification
 def sentiment_classification(summary):
  sentiment_model = pipeline("text-classification", model="wxrrrrrrr/finetunde_sentiment_analysis")
- result = sentiment_model(summary, max_length=100, do_sample=False)[0]['label']
+ result = sentiment_model(summary, max_length=100, truncation=True)[0]['label']
  return result
 
 def main():