Update app.py

app.py

@@ -1,12 +1,10 @@
 import gradio as gr
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import pipeline
 from PIL import Image
 import io
 
-# 🔍 تحميل النموذج والمُعالِج
-model_id = "HuggingFaceM4/sla-v1"
-processor = AutoProcessor.from_pretrained(model_id)
-model = AutoModelForVision2Seq.from_pretrained(model_id).to("cuda" if torch.cuda.is_available() else "cpu")
+# 📦 تحميل النموذج باستخدام Pipeline
+pipe = pipeline("image-to-text", model="microsoft/kosmos-2-patch14-224")
 
 # 💬 تعليمات الذكاء الاصطناعي لتكون خبير تداول
 SYSTEM_PROMPT = """
@@ -34,14 +32,11 @@ def analyze_chart(image):
         # إنشاء prompt شامل
         text = f"Question: {SYSTEM_PROMPT}\nAnswer:"
 
-        # المعالجة
-        inputs = processor(images=img, text=text, return_tensors="pt").to("cuda" if torch.cuda.is_available() else "cpu")
+        # التحليل باستخدام Pipeline
+        response = pipe(img, text)
 
-        # التوليد
-        generated_ids = model.generate(**inputs, max_new_tokens=256)
-        response = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-
-        return response.strip()
+        # تنظيف الإجابة
+        return response[0]["generated_text"]
 
     except Exception as e:
         return f"❌ حدث خطأ أثناء التحليل: {str(e)}"