Spaces:
Configuration error
Configuration error
Delete app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|