update upload function

app.py

@@ -22,26 +22,30 @@ categories = {
 }
 
 def prediction(news):
-    clasifer = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model, return_all_scores=True)
+    clasifer = pipeline("text-classification", tokenizer=tokenizer, model=model, return_all_scores=True)
     preds = clasifer(news)
     preds_dict = {categories.get(pred['label'], pred['label']): pred['score'] for pred in preds[0]}
     return preds_dict
 
 def file_prediction(file):
     if file.name.endswith('.csv'):
-        df = pd.read_csv(file) 
-        news_list = df.iloc[:, 0].tolist() 
+        df = pd.read_csv(file)
+        news_list = df.iloc[:, 0].tolist()
     else:
         file.seek(0)  
-        file_content = file.read().decode('utf-8')  
+        file_content = file.read().decode('utf-8')
         news_list = file_content.splitlines()
-    
+
     results = []
     for news in news_list:
         if news.strip():
-            results.append(prediction(news))
+            pred = prediction(news)
+            results.append(pred)
     
-    return pd.DataFrame(results, index=news_list)  
+    return pd.DataFrame({
+        'News': news_list,
+        'Prediction': results
+    })