edit predict func

app.py

@@ -10,7 +10,9 @@ classifier = pipeline(
 
 ############### Tạo giao diện ###############
 def predict(text):
-    return classifier(text)[0]
+    predictions = classifier(text)[0]
+    result = {pred['label']: pred['score'] for pred in predictions}
+    return result
 
 demo = gr.Interface(
     fn=predict,

requirements.txt

@@ -1,4 +1,6 @@
 transformers==4.35.0
 gradio==5.9.1
-torch==2.0.0
-tokenizers==0.14.1
+torch==2.6.0
+tokenizers==0.14.1
+tensorflow==2.13.0
+keras==2.13.1