Update app.py

app.py

@@ -55,7 +55,7 @@ else:
     raise ValueError(model_name)
 
 model = load_model_checkpoint(args=args, device="cpu")
-model.to("cuda")
+# model.to("cuda")
 # @title GradIO helper