Update app.py

app.py

@@ -2,6 +2,10 @@ import streamlit as st
 import random
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+# URL des Logos
+logo_url = "https://dein-logo-url-hier.com/logo.png"  # Ersetze dies mit der tatsächlichen URL Deines Logos
+
+# Modell und Tokenizer initialisieren
 tokenizer = AutoTokenizer.from_pretrained("Kaludi/chatgpt-gpt4-prompts-bart-large-cnn-samsum")
 model = AutoModelForSeq2SeqLM.from_pretrained("Kaludi/chatgpt-gpt4-prompts-bart-large-cnn-samsum", from_tf=True)
 
@@ -11,20 +15,22 @@ def generate(prompt, max_new_tokens):
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
     return output[0]
 
+# Streamlit App Layout
+st.markdown(f"<img src='{logo_url}' style='max-height: 100px;'>", unsafe_allow_html=True)  # Logo einfügen
 st.title("ChatGPT_Streamlit-Prompt Generator")
-st.write("This app generates ChatGPT/BingChat model trained by [Kaludi](https://huggingface.co/Kaludi/). Enter a role and a prompt will be generated based on it.")
-prompt = st.text_input("Enter a Role, Example: Virtual Assistant", placeholder="Text here", value="")
-max_new_tokens = st.slider("Select Max Tokens in Response", min_value=100, max_value=500, value=150, step=10)
-if st.button("Generate"):
+st.write("Gib eine Rolle ein, und es wird ein Prompt basierend darauf generiert.")
+prompt = st.text_input("Gib eine Rolle ein, Beispiel: Virtueller Assistent", placeholder="Text hier", value="")
+max_new_tokens = st.slider("Wähle die maximale Anzahl an Tokens in der Antwort", min_value=100, max_value=500, value=150, step=10)
+
+if st.button("Generieren"):
     output = generate(prompt, max_new_tokens)
-    st.write("Generated Prompt:", box=True)
-    st.write("<div style='background-color: #2E2E2E; padding: 10px;'>{}</div>".format(output), unsafe_allow_html=True)
-st.write("")
-st.write("<div style='text-align: center; font-weight: bold;'>Examples:</div>",unsafe_allow_html=True, box=True)
-st.write("<style> .stBox span { background-color: #2E2E2E; } </style>", unsafe_allow_html=True)
-with open("examples.txt", "r") as f:
+    st.write("Generierter Prompt:")
+    st.write(output)
+
+# Beispiele anzeigen
+st.write("Beispiele:")
+with open("examples.txt", "r") as f:  # Ersetze "examples.txt" mit dem tatsächlichen Pfad zu Deiner Beispieldatei
     examples = f.readlines()
-    random_examples = random.sample(examples, 5)
-    for example in random_examples:
-        example = example.strip()
-        st.write("<div style='background-color: #2E2E2E; padding: 10px; text-align: center;'>• {}</div>".format(example), unsafe_allow_html=True)
+random_examples = random.sample(examples, 5)
+for example in random_examples:
+    st.write(f"• {example.strip()}")