Update app.py

app.py

@@ -1,18 +1,28 @@
 import gradio as gr
-from transformers import pipeline
+import requests
 
-pipe = pipeline("image-to-text", model="deepseek-ai/DeepSeek-VL2-Small")
+API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-VL2-Small"
+API_TOKEN = "hf_..."  # Твой Hugging Face токен доступа сюда (можно сгенерить)
 
-def predict(image):
-    result = pipe(image)
-    return result[0]['generated_text'] if isinstance(result, list) else result
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def query(image):
+    buffered = image.convert("RGB")
+    buffered.save("temp.jpg", format="JPEG")
+
+    with open("temp.jpg", "rb") as f:
+        data = f.read()
+
+    response = requests.post(API_URL, headers=headers, data=data)
+    output = response.json()
+    return output[0]["generated_text"] if isinstance(output, list) else output
 
 iface = gr.Interface(
-    fn=predict,
+    fn=query,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="DeepSeek Image-to-Text Generator",
-    description="Описание изображений через DeepSeek-VL2-Small"
+    title="DeepSeek Image Captioning",
+    description="Генерация описания изображения через DeepSeek VL2"
 )
 
 if __name__ == "__main__":