Asya2025 commited on
Commit
327d34f
·
verified ·
1 Parent(s): 63a3437

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -2,18 +2,18 @@ import os
2
  import requests
3
  import gradio as gr
4
 
5
- # API URL для твоей модели на Hugging Face
6
  API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-VL2-Small"
7
 
8
- # Получаем токен из переменной окружения и убираем лишние пробелы/переносы
9
  API_TOKEN = os.getenv("My_token").strip()
10
 
11
- # Заголовки для запроса с авторизацией
12
  headers = {
13
  "Authorization": f"Bearer {API_TOKEN}"
14
  }
15
 
16
- # Функция для отправки изображения на сервер модели
17
  def query(image):
18
  buffered = image.convert("RGB")
19
  buffered.save("temp.jpg", format="JPEG")
@@ -21,10 +21,10 @@ def query(image):
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]:
@@ -46,3 +46,4 @@ iface = gr.Interface(
46
  # Запуск приложения
47
  if __name__ == "__main__":
48
  iface.launch(share=True)
 
 
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")
 
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]:
 
46
  # Запуск приложения
47
  if __name__ == "__main__":
48
  iface.launch(share=True)
49
+