Upload app.py

app.py

@@ -6,6 +6,7 @@ from datasets import Dataset
 import torch
 from flask import Flask, request, jsonify, render_template
 from threading import Thread
+import gradio as gr
 
 app = Flask(__name__)
 
@@ -118,7 +119,19 @@ def run_flask():
         tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
         model = None
         labels = []
-    app.run()
+    app.run(port=5000)
+
+# تشغيل Gradio
+def run_gradio():
+    def classify(text):
+        inputs = tokenizer(text, return_tensors="pt", padding="max_length", truncation=True)
+        outputs = model(**inputs)
+        predictions = torch.argmax(outputs.logits, dim=-1)
+        label = labels[predictions.item()]
+        return label
+
+    gr.Interface(fn=classify, inputs="text", outputs="text").launch(server_name="0.0.0.0", server_port=7860)
 
 if __name__ == '__main__':
     Thread(target=run_flask).start()
+    Thread(target=run_gradio).start()