juancamval commited on
Commit
62b7635
·
verified ·
1 Parent(s): e2145e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,9 +4,12 @@ from transformers import pipeline
4
 
5
  st.title("Testing Gemma")
6
  input_text = st.text_input("How can i help you: ")
 
 
7
  messages = [
8
  {"role": "user", "content": input_text},
9
  ]
10
  pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))
11
 
12
- st.write(pipe(messages))
 
 
4
 
5
  st.title("Testing Gemma")
6
  input_text = st.text_input("How can i help you: ")
7
+ process = st.button("ask")
8
+
9
  messages = [
10
  {"role": "user", "content": input_text},
11
  ]
12
  pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))
13
 
14
+ if process:
15
+ st.write(pipe(messages))