ramhemanth580 commited on
Commit
6cd0c97
·
verified ·
1 Parent(s): b77cf7d

Update app.py

Browse files

pipe = pipeline("text-generation", model="Qwen/Qwen2-1.5B-Instruct")

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import pipeline
8
  app = FastAPI()
9
 
10
  ## Initiate the text generation pipeline
11
- pipe = pipeline("text-generation", model="Qwen/Qwen2-1.5B-Instruct-GPTQ-Int8")
12
 
13
 
14
  @app.get("/")
 
8
  app = FastAPI()
9
 
10
  ## Initiate the text generation pipeline
11
+ pipe = pipeline("text-generation", model="Qwen/Qwen2-1.5B-Instruct")
12
 
13
 
14
  @app.get("/")