Update app.py

app.py

@@ -3,14 +3,14 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
 # Initialisierung des Modells und des Tokenizers
-tokenizer = GPT2Tokenizer.from_pretrained("Loewolf/GPT_1.5")
-model = GPT2LMHeadModel.from_pretrained("Loewolf/GPT_1.5")
+tokenizer = GPT2Tokenizer.from_pretrained("Loewolf/GPT_1.5-medium")
+model = GPT2LMHeadModel.from_pretrained("Loewolf/GPT_1.5-medium")
 
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
     attention_mask = torch.ones(input_ids.shape, dtype=torch.long)
 
-    max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 50
+    max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 70
     beam_output = model.generate(
         input_ids,
         attention_mask=attention_mask,
@@ -22,7 +22,7 @@ def generate_text(prompt):
         temperature=0.6,
         top_p=0.95,
         top_k=10,
-        repetition_penalty: 1.2,
+        
         do_sample=True,
         eos_token_id=tokenizer.eos_token_id,
         pad_token_id=tokenizer.eos_token_id 
@@ -31,13 +31,7 @@ def generate_text(prompt):
     text = tokenizer.decode(beam_output[0], skip_special_tokens=True)
     return text
 
-DESCRIPTION = """\
-#Löwolf GPT1 Chat
 
-<p>Es wird neues Löwolf GPT 1.5 verwendet.</p>
-
-Löwolf Chat verwendet immer das aktuelle GPT Modell von Löwolf Community!
-"""
 css = """
 h1 {
   text-align: center;