Bob-Potato commited on
Commit
5be7b48
·
verified ·
1 Parent(s): 65ea8ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,10 +2,10 @@ import gradio as gr
2
  import os
3
  import requests
4
 
5
- # Alege un model public HF (gratuit)
6
- MODEL_ID = "google/flan-t5-small" # poți folosi orice model HF
7
 
8
- HF_API_TOKEN = os.getenv("HF_API_TOKEN")
9
  HF_API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
10
  HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"} if HF_API_TOKEN else {}
11
 
@@ -25,7 +25,6 @@ def ask_ai(question):
25
  return f"Eroare la apel HF Inference API: {str(e)}"
26
 
27
  res = r.json()
28
- # modelele HF returnează listă cu generated_text
29
  if isinstance(res, list) and "generated_text" in res[0]:
30
  return res[0]["generated_text"]
31
  return str(res)
 
2
  import os
3
  import requests
4
 
5
+ # Model public gratuit pentru HF Inference API
6
+ MODEL_ID = "bigscience/bloom-560m"
7
 
8
+ HF_API_TOKEN = os.getenv("HF_API_TOKEN") # poate fi gol pentru public
9
  HF_API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
10
  HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"} if HF_API_TOKEN else {}
11
 
 
25
  return f"Eroare la apel HF Inference API: {str(e)}"
26
 
27
  res = r.json()
 
28
  if isinstance(res, list) and "generated_text" in res[0]:
29
  return res[0]["generated_text"]
30
  return str(res)