Add application file

app.py

@@ -20,8 +20,14 @@ import torch
 # Import the UI components
 from ui_components import CUSTOM_CSS, HEADER_HTML, DISCLAIMER_HTML
 
-# Check for GPU availability
-device = "cuda" if torch.cuda.is_available() else "cpu"
+
+if torch.cuda.is_available():
+    device = "cuda"
+    print("Using GPU")
+else:
+    device = "cpu"
+    print("Using CPU")
+
 
 TAB_STRUCTURE = [
     ("Visual Analysis", [
@@ -49,7 +55,7 @@ TAB_STRUCTURE = [
     ])
 ]
 
-@spaces.GPU
+
 def create_demo():
     with gr.Blocks(css=CUSTOM_CSS) as demo:
         gr.Markdown(HEADER_HTML)