Update app.py

app.py

@@ -45,6 +45,7 @@ def start_controller():
     print(controller_command)
     return subprocess.Popen(controller_command)
 
+@spaces.GPU
 def start_worker(model_path: str):
     print(f"Starting the model worker for the model {model_path}")
     model_path = 'qnguyen3/nanoLLaVA'
@@ -62,8 +63,7 @@ def start_worker(model_path: str):
         "--model-path",
         model_path,
         "--model-type",
-        "qwen1.5-0.5b",
-        "--use-flash-attn",
+        "qwen1.5-0.5b"
     ]
     print(worker_command)
     return subprocess.Popen(worker_command)