ingsimo commited on
Commit
ed03f9d
·
verified ·
1 Parent(s): 284d9e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -1,9 +1,16 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline('sentiment-analysis')
5
- text =st.text_area('dsfs')
 
 
 
6
 
7
- if text:
8
- out = pipe(text)
 
 
 
 
9
  st.json(out)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ messages = [
5
+ {"role": "user", "content": "Who are you?"},
6
+ ]
7
+ pipe = pipeline("text-generation", model="iGeniusAI/Italia-9B-Instruct-v0.1", trust_remote_code=True)
8
+ pipe(messages)
9
 
10
+
11
+ #pipe = pipeline('sentiment-analysis')
12
+ #text =st.text_area('dsfs')
13
+
14
+ #if text:
15
+ out = pipe(messages)
16
  st.json(out)