Asya2025 commited on
Commit
5d4d283
·
verified ·
1 Parent(s): 51b1404

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -50
app.py DELETED
@@ -1,50 +0,0 @@
1
- import os
2
- import requests
3
- import gradio as gr
4
-
5
- # Указываем URL для модели на Hugging Face, используя inference API
6
- API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-VL2-Small"
7
-
8
- # Токен API, убедись, что ты правильно установил переменную окружения
9
- API_TOKEN = os.getenv("My_token").strip()
10
-
11
- # Заголовки для запроса
12
- headers = {
13
- "Authorization": f"Bearer {API_TOKEN}"
14
- }
15
-
16
- # Функция для отправки запроса к API
17
- def query(image):
18
- buffered = image.convert("RGB")
19
- buffered.save("temp.jpg", format="JPEG")
20
-
21
- with open("temp.jpg", "rb") as f:
22
- data = f.read()
23
-
24
- # Отправка запроса на сервер Hugging Face для инференса
25
- response = requests.post(API_URL, headers=headers, files={"file": data})
26
-
27
- # Проверяем ответ от сервера
28
- if response.status_code == 200:
29
- output = response.json()
30
- if isinstance(output, list) and "generated_text" in output[0]:
31
- return output[0]["generated_text"]
32
- else:
33
- return "Ошибка: Неверный формат ответа от модели."
34
- else:
35
- return f"Ошибка запроса: {response.status_code} - {response.text}"
36
-
37
- # Интерфейс Gradio
38
- iface = gr.Interface(
39
- fn=query,
40
- inputs=gr.Image(type="pil"),
41
- outputs="text",
42
- title="DeepSeek Image Captioning",
43
- description="Генерация описания изображения через DeepSeek VL2"
44
- )
45
-
46
- # Запуск приложения
47
- if __name__ == "__main__":
48
- iface.launch()
49
-
50
-