namantjeaswi commited on
Commit
1ac30bc
·
1 Parent(s): 5db8677

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,11 +24,11 @@ model_id = "mistralai/Mistral-7B-v0.1"
24
 
25
 
26
 
27
- tokenizer = AutoTokenizer.from_pretrained(model_id)
28
- model = AutoModelForCausalLM.from_pretrained(model_id)
29
 
30
  # Create text generation pipeline
31
- pipe = pipeline(model = model, tokenizer = tokenizer)
32
 
33
 
34
  with st.form('my_form'):
 
24
 
25
 
26
 
27
+ #tokenizer = AutoTokenizer.from_pretrained(model_id)
28
+ #model = AutoModelForCausalLM.from_pretrained(model_id)
29
 
30
  # Create text generation pipeline
31
+ pipe = pipeline(model = model_id)
32
 
33
 
34
  with st.form('my_form'):