Update app.py

app.py

@@ -14,13 +14,14 @@ def generate_unique_feedback(text):
     # Generate a prompt for writing feedback
     prompt = f"Provide constructive feedback on the following creative writing piece:\n\n{text}\n\nFeedback:"
     
-    # Tokenize the input
-    inputs = tokenizer(prompt, return_tensors="pt")
+    # Tokenize the input with attention mask
+    inputs = tokenizer(prompt, return_tensors="pt", padding=True)
     
     # Attempt to generate unique feedback
     for _ in range(5):  # Try up to 5 times to get unique feedback
         with torch.no_grad():
-            outputs = model.generate(inputs.input_ids, max_length=300, do_sample=True, top_p=0.85, temperature=0.7)
+            # Use max_new_tokens instead of max_length
+            outputs = model.generate(inputs.input_ids, attention_mask=inputs.attention_mask, max_new_tokens=300, do_sample=True, top_p=0.85, temperature=0.7)
         
         # Decode the response
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -57,5 +58,5 @@ with gr.Blocks() as app:
     reset_button = gr.Button("Reset Feedback History")
     reset_button.click(fn=reset_history, outputs=reset_status)
 
-# Launch the app
-app.launch()
+# Launch the app with a public shareable link
+app.launch(share=True)