Update app.py

app.py

@@ -12,30 +12,21 @@ 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=130,  # Adjusted to generate shorter text
+        max_length=100,  # Adjusted to generate shorter text
         num_return_sequences=1, 
         temperature=0.7,  # Increased to add more randomness
-        top_k=130,  # Increased to allow a wider selection of words
+        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']
     
-    # Post-process to remove duplicate sentences
-    processed_text = remove_duplicate_sentences(generated_text)
-    
-    return processed_text
+    return generated_text
 
 # Define the Gradio interface
 interface = gr.Interface(
@@ -46,5 +37,5 @@ interface = gr.Interface(
     description="A model fine-tuned for generating text related to polymers."
 )
 
-# Launch the interface 
+# Launch the interface
 interface.launch()