Update app.py

app.py

@@ -129,12 +129,13 @@ outputs = model.generate(
 )
 
         # Decode generated text
-description = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-# Display the final generated description
-st.write("### Final Generated Description")
-st.write(description)
+        description = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+        # Display the final generated description
+        st.write("### Final Generated Description")
+        st.write(description)
+    else:
+        st.warning("No data available for the selected watch.")
 else:
     st.warning("Please select a brand.")