gouravbhadraDev commited on
Commit
fd64d36
·
verified ·
1 Parent(s): f8146bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def scrape(url: str = Query(...)):
36
  return ThreadResponse(question=question, replies=replies)
37
  return ThreadResponse(question="", replies=[])
38
 
39
- MODEL_NAME = "manycore-research/SpatialLM-Llama-1B"
40
 
41
  # Load the pipeline once at startup with device auto-mapping
42
  text_generator = pipeline(
 
36
  return ThreadResponse(question=question, replies=replies)
37
  return ThreadResponse(question="", replies=[])
38
 
39
+ MODEL_NAME = "microsoft/phi-2"
40
 
41
  # Load the pipeline once at startup with device auto-mapping
42
  text_generator = pipeline(