Update app.py

app.py

@@ -8,14 +8,9 @@ model = GPT2LMHeadModel.from_pretrained("Loewolf/GPT_1")
 
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
+    attention_mask = torch.ones(input_ids.shape, dtype=torch.bool)  # Erstelle eine Attention-Mask, die überall '1' ist
 
-    # Erstelle eine Attention-Mask, die überall '1' ist
-    attention_mask = torch.ones(input_ids.shape, dtype=torch.bool)
-
-    # Bestimmung der maximalen Länge
     max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 20
-
-    # Erzeugen von Text mit spezifischen Parametern
     beam_output = model.generate(
         input_ids,
         attention_mask=attention_mask,
@@ -32,20 +27,29 @@ def generate_text(prompt):
         eos_token_id=tokenizer.eos_token_id,  # EOS Token setzen
         pad_token_id=tokenizer.eos_token_id 
     )
-
+    
     text = tokenizer.decode(beam_output[0], skip_special_tokens=True)
     return text
 
-# Erstellung des Chatbot-Interface mit dem Titel "Löwolf Chat"
+css = """
+    body { font-family: Arial, sans-serif; }
+    .gradio_container { max-width: 700px; margin: auto; padding-top: 50px; }
+    .gradio_header { display: none; }
+    .gradio_input_box { border-radius: 10px; }
+    .gradio_output_box { border-radius: 10px; }
+    button { background-color: #29B3FF; color: white; padding: 10px 20px; border: none; border-radius: 5px; cursor: pointer; }
+    button:hover { background-color: #106ba3; }
+"""
+
 iface = gr.Interface(
     fn=generate_text,
     inputs=gr.Textbox(label="Schreibe hier...", placeholder="Stelle deine Frage..."),
-    outputs="text",
+    outputs=gr.Textbox(label="Antwort"),
     title="Löwolf Chat",
-    description="Willkommen beim Löwolf Chat. Stelle deine Fragen und erhalte Antworten vom KI-Chatbot."
+    description="Willkommen beim Löwolf Chat. Stelle deine Fragen und erhalte Antworten vom KI-Chatbot.",
+    css=css
 )
 
-# Starten des Chatbot-Interfaces
 iface.launch()