trying new approach

Dockerfile

@@ -35,9 +35,8 @@ RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
 # It takes arguments like --model_path, --image_path, --prompt, --device
 # We will run it directly. Its output will go to the Space's container logs.
 CMD ["python", "-u", "/app/nanoVLM/generate.py", \
-     "--model_path", "lusxvr/nanoVLM-222M", \
-     "--image_path", "/app/test_image.jpg", \
+     "--hf_model", "lusxvr/nanoVLM-222M", \
+     "--image", "/app/test_image.jpg", \
      "--prompt", "describe this image in detail", \
-     "--device", "cpu", \
-     "--num_generations", "1", \
+     "--generations", "1", \
      "--max_new_tokens", "50"]