Anne314159's picture
Create app.py
dee81af verified
raw
history blame
973 Bytes
import streamlit as st
from transformers import pipeline
# Initialize a text generation pipeline
generator = pipeline('text-generation', model='dbmdz/german-gpt2')
st.title('German Medical Content Manager')
# Sidebar for user input
st.sidebar.header('User Input Options')
input_topic = st.sidebar.text_input('Enter a medical topic', 'Type 1 Diabetes')
# Main Page
st.write(f"Creating social media content for: {input_topic}")
# Function to generate text based on the input topic
def generate_content(topic):
generated_text = generator(f"Letzte Nachrichten über {topic}:", max_length=50, num_return_sequences=1)
return generated_text[0]['generated_text']
if st.button('Generate Social Media Post'):
with st.spinner('Generating...'):
post_content = generate_content(input_topic)
st.success('Generated Content:')
st.write(post_content)
st.write('Generated social media posts will appear here after clicking the "Generate" button.')