Update app.py

app.py

@@ -1,5 +1,5 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
-
+import streamlit as st
 device = "cuda" # the device to load the model onto
 
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
@@ -18,4 +18,4 @@ model.to(device)
 
 generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
 decoded = tokenizer.batch_decode(generated_ids)
-print(decoded[0])
+st.write(decoded[0])