Update app.py

app.py

@@ -10,10 +10,10 @@ def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
 
     # Erstelle eine Attention-Mask, die überall '1' ist
-    attention_mask = torch.ones(input_ids.shape, dtype=torch.bool)
+    attention_mask = torch.ones(input_ids.shape, dtype=torch.long)
 
     # 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
+    max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 100
 
     # Erzeugen von Text mit spezifischen Parametern
     beam_output = model.generate(