Update app.py

app.py

@@ -188,20 +188,20 @@ def improve_summary_generation(text, model, tokenizer):
     # Validate the summary
     if not validate_summary(processed_summary, text):
     # Retry with alternate generation parameters
-    with torch.no_grad():
-        summary_ids = model.generate(
-            input_ids=inputs["input_ids"],
-            attention_mask=inputs["attention_mask"],
-            max_length=250,
-            min_length=50,
-            num_beams=4,
-            length_penalty=2.0,
-            no_repeat_ngram_size=4,
-            temperature=0.8,
-            repetition_penalty=1.5,
-        )
-    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-    processed_summary = post_process_summary(summary)
+        with torch.no_grad():
+            summary_ids = model.generate(
+                input_ids=inputs["input_ids"],
+                attention_mask=inputs["attention_mask"],
+                max_length=250,
+                min_length=50,
+                num_beams=4,
+                length_penalty=2.0,
+                no_repeat_ngram_size=4,
+                temperature=0.8,
+                repetition_penalty=1.5,
+            )
+        summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+        processed_summary = post_process_summary(summary)
 
     
     return processed_summary