Update app.py

app.py

@@ -28,7 +28,6 @@
 
 import gradio as gr
 import spaces
-import torch
 import os
 
 # Global model variable, but do not initialize or move to CUDA here
@@ -39,6 +38,8 @@ def generate(model_name, image, text):
     global model_pipe
     
     # Debugging lines to trace CUDA initialization
+    import torch
+    
     print(f"PID: {os.getpid()}")
     print(f"Before import: CUDA available: {torch.cuda.is_available()}")