Hugging Face's logo

Spaces:
gouravbhadraDev
/
server_test
Running

App Files Community
gouravbhadraDev commited on
Commit
9f70441
·
verified ·
1 Parent(s): d3e9fb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def scrape(url: str = Query(...)):
35
  return ThreadResponse(question="", replies=[])
36
 
37
 
38
- MODEL_NAME = "mistralai/Magistral-Small-2506"
39
 
40
  # Load tokenizer and model once at startup, with device auto-mapping
41
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
35
  return ThreadResponse(question="", replies=[])
36
 
37
 
38
+ MODEL_NAME = "google/flan-t5-small"
39
 
40
  # Load tokenizer and model once at startup, with device auto-mapping
41
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)