DanilNikolin commited on
Commit
142c205
·
verified ·
1 Parent(s): 34e8492

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  import requests
3
  import os
4
 
5
- API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-2.7B"
6
- API_KEY = os.getenv('Sec1') # Используем переменную окружения
7
 
8
  def query(payload):
9
  response = requests.post(API_URL, headers={"Authorization": f"Bearer {API_KEY}"}, json=payload)
@@ -11,11 +11,7 @@ def query(payload):
11
 
12
  def chatbot(input_text):
13
  response = query({"inputs": input_text})
14
- print(response) # Выводим ответ от API в консоль для отладки
15
- try:
16
- return response[0]['generated_text']
17
- except (KeyError, IndexError):
18
- return "Error: Unable to generate text. Please try again."
19
 
20
  demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
21
  demo.launch()
 
2
  import requests
3
  import os
4
 
5
+ API_URL = "https://api-inference.huggingface.co/models/YOUR_GEMINI_MODEL"
6
+ API_KEY = os.getenv('YOUR_HUGGINGFACE_TOKEN')
7
 
8
  def query(payload):
9
  response = requests.post(API_URL, headers={"Authorization": f"Bearer {API_KEY}"}, json=payload)
 
11
 
12
  def chatbot(input_text):
13
  response = query({"inputs": input_text})
14
+ return response[0]['generated_text']
 
 
 
 
15
 
16
  demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
17
  demo.launch()