update app.py

app.py

@@ -28,8 +28,8 @@ model_id = "mistralai/Mistral-7B-v0.1"
 #model = AutoModelForCausalLM.from_pretrained(model_id)
 
 # Create text generation pipeline
-pipe = pipeline(model = model_id)
-
+#pipe = pipeline(model = model_id)
+pipe = pipeline('text-generation')
 
 with st.form('my_form'):
     question = st.text_area('Enter your question:', 'Tell me about attention mechanisms in a transformer?')
@@ -37,4 +37,4 @@ with st.form('my_form'):
     if submitted:
         result = pipe(question, max_length=100)
         st.write(question)
-        st.write(result[0]['generated_text'])
+        st.write(result)