Spaces:

syedfaisalabrar
/

License_Classification

Sleeping

syedfaisalabrar commited on Feb 26

Commit

07a53e4

verified ·

1 Parent(s): 1291d1b

Update backPrompt.py

Files changed (1) hide show

backPrompt.py CHANGED Viewed

@@ -27,7 +27,7 @@ def load_image(image_file):
 def main(image_path,model,tokenizer):
-    path = "OpenGVLab/InternVL2_5-4B"
     pixel_values = load_image(image_path).to(torch.float32).to("cpu")
     generation_config = dict(max_new_tokens=1024, do_sample=True)

 def main(image_path,model,tokenizer):
     pixel_values = load_image(image_path).to(torch.float32).to("cpu")
     generation_config = dict(max_new_tokens=1024, do_sample=True)