AminFaraji commited on
Commit
a37d6d4
·
verified ·
1 Parent(s): a11cce5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ MODEL_NAME = "tiiuae/falcon-7b-instruct"
51
 
52
  llama_pipeline = pipeline(
53
  "text-generation",
54
- model=model,
55
  torch_dtype=torch.float16,
56
  device_map="auto",
57
  )
 
51
 
52
  llama_pipeline = pipeline(
53
  "text-generation",
54
+ model=MODEL_NAME,
55
  torch_dtype=torch.float16,
56
  device_map="auto",
57
  )