Spaces:

witcher23
/

nanoVLM-inference

Running

vidhanm commited on 25 days ago

Commit

f2f290e

1 Parent(s): 08a37b7

reducing tokens to 20

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def call_generate_script(image_path: str, prompt_text: str) -> str:
         "--image", image_path,
         "--prompt", prompt_text,
         "--generations", "1",      # Get one generation for the UI
-        "--max_new_tokens", "70"   # Adjust as needed
         # --device is handled by generate.py internally
     ]

         "--image", image_path,
         "--prompt", prompt_text,
         "--generations", "1",      # Get one generation for the UI
+        "--max_new_tokens", "20"   # Adjust as needed
         # --device is handled by generate.py internally
     ]