Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the model
-model = pipeline("text-generation", model="rish13/polymers")
+# Load the model with optimization settings
+model = pipeline(
+    "text-generation",
+    model="rish13/polymers",
+    device=0  # Use device=0 for GPU (if available), -1 for CPU
+)
 
 def generate_response(prompt):
     # Generate text from the model
-    response = model(prompt, max_length=150, num_return_sequences=1)
+    response = model(prompt, max_length=100, num_return_sequences=1, temperature=0.7)
     
     # Get the generated text from the response
     generated_text = response[0]['generated_text']