Update app.py

app.py

@@ -47,9 +47,10 @@ if selected_brand != "Select":
 
     if watch_data:
         # Generation parameters
-        max_length = st.slider("Max Length", min_value=50, max_value=300, value=150)
+        max_length = st.slider("Max Length", min_value=100, max_value=500, value=300)  # Increase max length to allow for more text
         temperature = st.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
-        
+        repetition_penalty = st.slider("Repetition Penalty", min_value=1.0, max_value=2.0, value=1.2, step=0.1)  # Penalize repetitive text
+
         # Generate description based on attributes
         if st.button("Generate Description"):
             attributes = {
@@ -61,7 +62,7 @@ if selected_brand != "Select":
                 "movement": watch_data.get("movement", "Unknown Movement"),
                 "gender": watch_data.get("gender", "Unknown Gender"),
             }
-            
+
             # Format input similar to training data (adjust this based on your training data format)
             input_text = f"""Generate a detailed description for the following watch:
 Brand: {attributes['brand']}
@@ -73,26 +74,27 @@ Movement: {attributes['movement']}
 Gender: {attributes['gender']}
 
 Description:"""
-            
+
             # Tokenize input and generate description
             inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
             outputs = model.generate(
-                **inputs, 
-                max_length=max_length, 
+                **inputs,
+                max_length=max_length,
                 num_return_sequences=1,
                 temperature=temperature,
                 top_k=50,
                 top_p=0.95,
-                do_sample=True
+                do_sample=True,
+                repetition_penalty=repetition_penalty  # Prevent repetitive descriptions
             )
-            
+
             # Decode generated text
             description = tokenizer.decode(outputs[0], skip_special_tokens=True)
-            
+
             # Display the result
             st.write("### Generated Description")
             st.write(description)
-            
+
             # Display watch details
             st.write("### Watch Details")
             st.json(json.dumps(watch_data, indent=2))
@@ -130,4 +132,4 @@ st.markdown(
     </div>
     """,
     unsafe_allow_html=True
-)
+)