Update app.py

app.py

@@ -8,14 +8,14 @@ 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
+    attention_mask = torch.ones(input_ids.shape, dtype=torch.bool)
 
     max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 20
     beam_output = model.generate(
         input_ids,
         attention_mask=attention_mask,
         max_length=max_length,
-        min_length=4,  # Mindestlänge der Antwort
+        min_length=4,
         num_beams=5,
         no_repeat_ngram_size=2,
         early_stopping=True,
@@ -24,7 +24,7 @@ def generate_text(prompt):
         top_k=50,
         length_penalty=2.0,
         do_sample=True,
-        eos_token_id=tokenizer.eos_token_id,  # EOS Token setzen
+        eos_token_id=tokenizer.eos_token_id,
         pad_token_id=tokenizer.eos_token_id 
     )
     
@@ -32,24 +32,24 @@ def generate_text(prompt):
     return text
 
 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; }
+    body { font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif; }
+    .gradio_app { max-width: 600px; margin: 50px auto; }
+    .gradio_interface { box-shadow: 0 0 20px rgba(0,0,0,0.1); }
+    .gradio_input_container { width: 100%; }
+    .gradio_output_container { width: 100%; margin-bottom: 25px; }
+    .gradio_input { border-radius: 20px; }
+    .gradio_output { border-radius: 20px; }
+    button { border-radius: 20px; }
 """
 
 iface = gr.Interface(
     fn=generate_text,
-    inputs=gr.Textbox(label="Schreibe hier...", placeholder="Stelle deine Frage..."),
-    outputs=gr.Textbox(label="Antwort"),
-    title="Löwolf Chat",
-    description="Willkommen beim Löwolf Chat. Stelle deine Fragen und erhalte Antworten vom KI-Chatbot.",
+    inputs=gr.Textbox(placeholder="Type a message..."),
+    outputs="text",
     css=css
 )
 
 iface.launch()
 
 
+