Update app.py

app.py

@@ -8,10 +8,10 @@ import re
 from torchvision.transforms.v2 import Resize
 
 parser = argparse.ArgumentParser()
-parser.add_argument("--cpu", action="store_true")
-args = parser.parse_args()
+parser.add_argument("--cpu", action="store_true", help="Use CPU for computation")
+args = parser.parse_args([])
 
-DEVICE = "cuda"
+DEVICE = "cuda" if torch.cuda.is_available() and not args.cpu else "cpu" # Determine device based on availability and argument
 DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
 LATEST_REVISION = "2024-05-20"
 
@@ -78,7 +78,7 @@ def process_answer(img, answer):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        # 🌔 moondream
+        # 🌔 moondream-finetuned
         """
     )
     with gr.Row():
@@ -94,4 +94,5 @@ with gr.Blocks() as demo:
     prompt.submit(answer_question, [img, prompt], output)
     output.change(process_answer, [img, output], ann, show_progress=False)
 
-demo.queue().launch(debug=True)
+
+demo.queue().launch(debug=False)