Rooni commited on
Commit
48a4b65
1 Parent(s): 849dbf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from PIL import Image
3
  import requests
4
  from io import BytesIO
 
5
 
6
  # Функция для взаимодействия с моделью на Hugging Face
7
  def process_image(image, prompt):
@@ -10,7 +11,7 @@ def process_image(image, prompt):
10
 
11
  # Отправка запроса на API Hugging Face
12
  headers = {
13
- "Authorization": f"Bearer YOUR_HUGGINGFACE_TOKEN" # Замените YOUR_HUGGINGFACE_TOKEN на ваш токен
14
  }
15
  data = {
16
  "inputs": {
 
2
  from PIL import Image
3
  import requests
4
  from io import BytesIO
5
+ import os
6
 
7
  # Функция для взаимодействия с моделью на Hugging Face
8
  def process_image(image, prompt):
 
11
 
12
  # Отправка запроса на API Hugging Face
13
  headers = {
14
+ "Authorization": f"Bearer {os.getenv('HF_TOKEN')}" # Использование API ключа из переменной окружения
15
  }
16
  data = {
17
  "inputs": {