Update app.py

app.py

@@ -57,10 +57,10 @@ def initialize_model():
         token=token,
         device_map="cuda"
     )
-    if torch.cuda.is_available():
-        model.device = "cuda"
-    else:
-        print("CUDA is not available")
+    # if torch.cuda.is_available():
+    #     model.device = "cuda"
+    # else:
+    #     print("CUDA is not available")
     
     return model, tokenizer