slfagrouche commited on
Commit
9d0fe26
·
1 Parent(s): 6db7691

Add application file

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -1,3 +1,21 @@
1
  import streamlit as st
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
- st.markdown("Hello world!")
 
1
  import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # Load the text generation pipeline
5
+ generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")
6
+
7
+ # Streamlit app title and description
8
+ st.title("Gemma Text Generation App")
9
+ st.write("This app generates text based on the input prompt using the Gemma model.")
10
+
11
+ # Text input for user prompt
12
+ prompt = st.text_input("Enter your prompt:", "Once upon a time,")
13
+
14
+ # Generate button to trigger text generation
15
+ if st.button("Generate Text"):
16
+ # Generate text based on the user prompt
17
+ generated_text = generator(prompt, max_length=100)[0]['generated_text']
18
+ # Display the generated text
19
+ st.write("Generated Text:")
20
+ st.write(generated_text)
21