update upload function

app.py

@@ -24,7 +24,7 @@ categories = {
 def prediction(news):
     classifier = pipeline("text-classification", tokenizer=tokenizer, model=model, return_all_scores=True)
     preds = classifier(news)
-    preds_dict = {categories.get(pred['label'], pred['label']): pred['score'] for pred in preds[0]}
+    preds_dict = {categories.get(pred['label'], pred['label']): round(pred['score'], 4) for pred in preds[0]}
     return preds_dict
 
 def file_prediction(file):
@@ -41,11 +41,11 @@ def file_prediction(file):
 
     results = []
     for news in news_list:
-        if news.strip():
+        if news.strip(): 
             pred = prediction(news)
-            results.append([news, pred])  # Return each news and its prediction as a row
+            results.append([news, pred])
 
-    return results  # Gradio expects a list of lists or dicts for DataFrame
+    return results
 
 gradio_ui = gr.Interface(
     fn=prediction,
@@ -69,3 +69,4 @@ gradio_file_ui = gr.Interface(
 gradio_combined_ui = gr.TabbedInterface([gradio_ui, gradio_file_ui], ["Text Input", "File Upload"])
 
 gradio_combined_ui.launch()
+