Update app.py

app.py

@@ -40,18 +40,17 @@ def generate_text(model, tokenizer, prompt, max_len, temperature):
     #                         no_repeat_ngram_size=3,
     #                         num_return_sequences=1,
     #                        pad_token_id=tokenizer.eos_token_id)
-        output = model.generate(
-        input_ids,
-        max_length=max_length,
-        temperature=temperature,              # Controls the diversity of the generated text
-        top_k=50,                     # Keeps only the top-k most likely words
-        top_p=0.9,                    # Nucleus sampling (cumulative probability)
-        repetition_penalty=1.2,       # Penalty for repeating words or phrases
-        no_repeat_ngram_size=4,       # Prevents repetition of n-grams (e.g., bigrams)
-        do_sample=True,                # Enables sampling for greater diversity
-        pad_token_id=tokenizer.eos_token_id,
-        max_length=max_len
-    )[0]
+    output = model.generate(
+    input_ids,
+    max_length=max_length,
+    temperature=temperature,              # Controls the diversity of the generated text
+    top_k=50,                     # Keeps only the top-k most likely words
+    top_p=0.9,                    # Nucleus sampling (cumulative probability)
+    repetition_penalty=1.2,       # Penalty for repeating words or phrases
+    no_repeat_ngram_size=4,       # Prevents repetition of n-grams (e.g., bigrams)
+    do_sample=True,                # Enables sampling for greater diversity
+    pad_token_id=tokenizer.eos_token_id,
+    max_length=max_len)[0]
 
 
     #generated_text = tokenizer.decode(output[0], skip_special_tokens=True)