Commit
·
6d6488b
1
Parent(s):
e7ca75f
update app.py
Browse files
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?')
|