Update app.py

app.py

@@ -12,21 +12,30 @@ model = pipeline(
     device=device  # Automatically use GPU if available, otherwise CPU
 )
 
+def remove_duplicate_sentences(text):
+    # Split text into sentences (assuming sentences are separated by periods)
+    sentences = text.split('. ')
+    unique_sentences = list(dict.fromkeys(sentences))  # Remove exact duplicates
+    return '. '.join(unique_sentences)
+
 def generate_response(prompt):
     # Generate text from the model
     response = model(
         prompt, 
         max_length=50,  # Adjusted to generate shorter text
         num_return_sequences=1, 
-        temperature=0.5,  # Lowered to reduce randomness
-        top_k=50,  # Limiting the next word selection
-        top_p=0.9  # Cumulative probability threshold
+        temperature=0.6,  # Increased to add more randomness
+        top_k=100,  # Increased to allow a wider selection of words
+        top_p=0.95  # Slightly increased cumulative probability threshold
     )
     
     # Get the generated text from the response
     generated_text = response[0]['generated_text']
     
-    return generated_text
+    # Post-process to remove duplicate sentences
+    processed_text = remove_duplicate_sentences(generated_text)
+    
+    return processed_text
 
 # Define the Gradio interface
 interface = gr.Interface(