Update app.py

app.py

@@ -18,15 +18,19 @@ user_input = st.text_input("Enter your message:")
 # Preprocess and generate response when the user hits Enter
 if user_input:
     if user_input.lower() == "quit":
-        st.stop()  # End the Streamlit app
+        st.stop()
 
-    encoded_input = tokenizer(user_input, return_tensors='pt')
-    outputs = model(**encoded_input)
-    logits = outputs.logits
-    predicted_class_id = logits.argmax(-1).item()
+    # Encode the user input 
+    input_ids = tokenizer.encode(user_input, return_tensors='pt') 
 
-    
-    sentiment_map = {0: "negative", 1: "neutral", 2: "positive"}  # Sentiment mapping
-    sentiment = sentiment_map[predicted_class_id]  # Get predicted sentiment 
+    # Generate a response (adjust parameters for control)
+    output_sequences = model.generate(
+        input_ids=input_ids,
+        max_length=50,  # Example max response length
+        temperature=0.8,  # Controls creativity
+        # ... other generation parameters ...
+    )
 
-    st.write(f"Predicted Sentiment: {sentiment}")  # Display the sentiment 
+    # Decode the generated text and display
+    generated_text = tokenizer.decode(output_sequences[0], skip_special_tokens=True)  
+    st.write(generated_text)