Update app.py

app.py

@@ -195,8 +195,8 @@ Now format the following:
             truncation=True,
             max_length=2048
         )
-        inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
-
+        #inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
+        inputs = {k: v for k, v in inputs.items()}
         with torch.no_grad():
             outputs = self.model.generate(
                 **inputs,
@@ -428,7 +428,7 @@ Speaker 2: ...
             
             # Move to correct device
             inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
-            add_log(f"✅ Inputs moved to device: {self.model.device}")
+            add_log(f"✅ Inputs moved to device: ")
 
             add_log("self🧠 Generating with model...")