Update app.py

app.py

@@ -2,17 +2,17 @@ import torch
 import transformers
 from transformers import AutoModelForCausalLM
 
-# Load the FastChat model
+
 model = AutoModelForCausalLM.from_pretrained("lmsys/fastchat-t5-3b-v1.0")
 
-# Start the conversation loop
+
 while True:
 
-    # Get the user input
+    
     user_input = input("You: ")
 
-    # Generate a response from the model
+    
     model_response = model.generate(input_ids=torch.tensor([model.bos_token_id]).unsqueeze(0), prompt=user_input)
 
-    # Print the response to the user
+    
     print("Assistant:", model_response[0][1:])