Update app.py

app.py

@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained("adi2606/MenstrualQA")
 
 # Function to generate a response from the chatbot
 def generate_response(message: str, temperature: float = 0.4, repetition_penalty: float = 1.1) -> str:
-    inputs = tokenizer(message, return_tensors="pt", padding=True, truncation=True).to(device)
+    inputs = tokenizer(message, return_tensors="pt", padding=True, truncation=True)
 
     # Generate the response
     output = model.generate(