Update app.py

app.py

@@ -1,8 +1,9 @@
 import joblib
 import gradio as gr
 from huggingface_hub import hf_hub_download
+import spaces
 
-# تحميل الملفات من مستودعك على Hugging Face مباشرة
+# تحميل الملفات من مستودعك على Hugging Face
 repo_id = "zico2m/comments-classifiers"
 
 vectorizer_path = hf_hub_download(repo_id=repo_id, filename="tfidf_vectorizer.pkl")
@@ -14,7 +15,8 @@ vectorizer = joblib.load(vectorizer_path)
 model = joblib.load(model_path)
 sentiment_model = joblib.load(sentiment_path)
 
-# دالة التنبؤ والتحليل
+# استخدام @spaces.GPU حتى لو كان النموذج يعمل على الـ CPU، لترضى المنصة بالتشغيل
+@spaces.GPU
 def classify_comment(text):
     if not text.strip():
         return "الرجاء إدخال نص صالح للتحليل."
@@ -22,7 +24,7 @@ def classify_comment(text):
     # تحويل النص باستخدام TF-IDF
     transformed_text = vectorizer.transform([text])
     
-    # التنبؤ (تعديل الأسماء حسب التصنيفات الموجودة عندك في النموذج)
+    # التنبؤ
     prediction = model.predict(transformed_text)[0]
     sentiment = sentiment_model.predict(transformed_text)[0]