Update app.py

app.py

@@ -120,7 +120,7 @@ def generate(
         torch_dtype = torch.float16
     )
     
-    pipe.to(device)
+   # pipe.to(device)
 
     try:
 
@@ -140,6 +140,12 @@ def generate(
     except Exception as e:
         print(f"An error occurred: {e}")
 
+if torch.cuda.is_available():
+    pipe = load_pipeline(MODEL)
+    print("Loaded on Device!")
+else:
+    pipe = None
+
 with gr.Blocks(
     theme=gr.themes.Soft()
 ) as demo: