Update app.py

app.py

@@ -1,9 +1,16 @@
 import streamlit as st
 from transformers import pipeline
 
-pipe = pipeline('sentiment-analysis')
-text =st.text_area('dsfs')
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="iGeniusAI/Italia-9B-Instruct-v0.1", trust_remote_code=True)
+pipe(messages)
 
-if text:
-    out = pipe(text)
+
+#pipe = pipeline('sentiment-analysis')
+#text =st.text_area('dsfs')
+
+#if text:
+    out = pipe(messages)
     st.json(out)