namantjeaswi commited on
Commit
6d6488b
·
1 Parent(s): e7ca75f

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ model_id = "mistralai/Mistral-7B-v0.1"
29
 
30
  # Create text generation pipeline
31
  #pipe = pipeline(model = model_id)
32
- pipe = pipeline('text-generation')
33
 
34
  with st.form('my_form'):
35
  question = st.text_area('Enter your question:', 'Tell me about attention mechanisms in a transformer?')
 
29
 
30
  # Create text generation pipeline
31
  #pipe = pipeline(model = model_id)
32
+ pipe = pipeline('text-generation', model=model_id)
33
 
34
  with st.form('my_form'):
35
  question = st.text_area('Enter your question:', 'Tell me about attention mechanisms in a transformer?')