Alfasign commited on
Commit
4dd0859
1 Parent(s): 6bc996a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -1,14 +1,15 @@
1
-
2
  from transformers import pipeline
 
3
 
4
  # Erstelle ein prompt generator
5
  generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
6
 
7
- # Frage den Benutzer nach einem Satzanfang
8
- prompt = input('Gib einen Satzanfang ein: ')
9
-
10
- # Generiere einen neuen Text basierend auf dem Satzanfang
11
- text = generator(prompt, max_length=30, do_sample=True, temperature=0.7)[0]['generated_text']
12
 
13
- # Gib den generierten Text aus
14
- print(text)
 
 
 
 
1
  from transformers import pipeline
2
+ import streamlit as st
3
 
4
  # Erstelle ein prompt generator
5
  generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
6
 
7
+ # Definiere die Streamlit-Oberfläche
8
+ st.title("Prompt Generator")
9
+ prompt = st.text_input("Gib einen Satzanfang ein:")
10
+ button = st.button("Generieren")
 
11
 
12
+ # Generiere neuen Text beim Klick auf den Button
13
+ if button:
14
+ text = generator(prompt, max_length=30, do_sample=True, temperature=0.7)[0]['generated_text']
15
+ st.write(text)