Hugging Face's logo

Spaces:
tyagonzales66
/
DiscussionBot-Webhook
Sleeping

App Files Community
tyagonzales66 commited on
Commit
84e8f63
·
verified ·
1 Parent(s): 474c23c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ app = FastAPI()
7
 
8
  # Initialisation du pipeline pour l'API d'inférence Hugging Face
9
  BOT_USERNAME = "@DiscussionBot"
10
- MODEL = "meta-llama/Llama-3.2-1B-Instruct"
11
  try:
12
  pipe = pipeline("text-generation", model=MODEL, token=os.getenv("HF_TOKEN"))
13
  except Exception as e:
 
7
 
8
  # Initialisation du pipeline pour l'API d'inférence Hugging Face
9
  BOT_USERNAME = "@DiscussionBot"
10
+ MODEL = "gpt2"
11
  try:
12
  pipe = pipeline("text-generation", model=MODEL, token=os.getenv("HF_TOKEN"))
13
  except Exception as e: