Update app.py

app.py

@@ -14,33 +14,43 @@ model = pipeline(
 
 def generate_response(prompt):
     # Generate text from the model
-    response = model(prompt, max_length=100, num_return_sequences=1, temperature=0.7)
+    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
+    )
     
     # Get the generated text from the response
     generated_text = response[0]['generated_text']
-
-    # Find the position of the first end-of-sentence punctuation
-    end_punctuation = ['.', '!', '?']
-    end_position = -1
-    for punct in end_punctuation:
-        pos = generated_text.find(punct)
-        if pos != -1 and (end_position == -1 or pos < end_position):
-            end_position = pos
-
-    # If punctuation is found, truncate the text at that point
-    if end_position != -1:
-        generated_text = generated_text[:end_position + 1]
     
     return generated_text
 
 # Define the Gradio interface
 interface = gr.Interface(
     fn=generate_response,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
-    outputs="text",
+    inputs=gr.Textbox(
+        lines=2, 
+        placeholder="Enter your prompt here...", 
+        label="Prompt",
+        elem_id="input-textbox"  # Custom styling ID for input textbox
+    ),
+    outputs=gr.Textbox(
+        label="Generated Text",
+        elem_id="output-textbox"  # Custom styling ID for output textbox
+    ),
     title="Polymer Knowledge Model",
-    description="A model fine-tuned for generating text related to polymers."
+    description=(
+        "This application uses a fine-tuned model to generate text related to polymers. "
+        "Enter a prompt to get started, and the model will generate relevant text."
+    ),
+    theme="huggingface",  # Apply a theme for consistent styling
+    layout="horizontal",  # Arrange input and output side by side
+    live=True  # Update the output live as the user types
 )
 
 # Launch the interface
 interface.launch()
+