Update app.py

app.py

@@ -300,15 +300,21 @@ with col2:
             AI:"""
             
         input_ids = tokenizer.encode(prompt, return_tensors="pt")
-            output = model.generate(
-                input_ids, max_length=300, num_return_sequences=1,
-                no_repeat_ngram_size=2, top_k=50, top_p=0.95, temperature=0.7
-            )
-            
-            response = tokenizer.decode(output[0], skip_special_tokens=True).split("AI:")[-1].strip()
-            
-            st.write("### AI's Sensory Analysis:")
-            st.write(response)
+
+output = model.generate(
+    input_ids, 
+    max_length=300, 
+    num_return_sequences=1,
+    no_repeat_ngram_size=2, 
+    top_k=50, 
+    top_p=0.95, 
+    temperature=0.7
+)
+
+response = tokenizer.decode(output[0], skip_special_tokens=True).split("AI:")[-1].strip()
+
+st.write("### AI's Sensory Analysis:")
+st.write(response)
 
 # Visualize sensation map
 st.subheader("Quantum Neuro-Sensory Map")