Rooni commited on
Commit
bf97a50
1 Parent(s): e045987

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  import requests
3
  import os
 
 
4
 
5
  # Функция для генерации текста
6
  def generate_text(prompt, model_choice, max_tokens, *other_params):
@@ -14,7 +16,7 @@ def generate_text(prompt, model_choice, max_tokens, *other_params):
14
 
15
  # Подготовка данных для запроса
16
  headers = {
17
- "Authorization": f"Bearer {os.getenv("HF_TOKEN")}"
18
  }
19
  payload = {
20
  "inputs": prompt,
 
1
  import gradio as gr
2
  import requests
3
  import os
4
+
5
+ token = os.getenv("HF_TOKEN")
6
 
7
  # Функция для генерации текста
8
  def generate_text(prompt, model_choice, max_tokens, *other_params):
 
16
 
17
  # Подготовка данных для запроса
18
  headers = {
19
+ "Authorization": f"Bearer {token}"
20
  }
21
  payload = {
22
  "inputs": prompt,