Update backPrompt.py
Browse files- backPrompt.py +1 -1
backPrompt.py
CHANGED
|
@@ -28,7 +28,7 @@ def load_image(image_file, input_size=800, max_num=12):
|
|
| 28 |
return pixel_values
|
| 29 |
|
| 30 |
|
| 31 |
-
def main(image_path
|
| 32 |
path = "OpenGVLab/InternVL2_5-4B"
|
| 33 |
model = AutoModel.from_pretrained(
|
| 34 |
path,
|
|
|
|
| 28 |
return pixel_values
|
| 29 |
|
| 30 |
|
| 31 |
+
def main(image_path):
|
| 32 |
path = "OpenGVLab/InternVL2_5-4B"
|
| 33 |
model = AutoModel.from_pretrained(
|
| 34 |
path,
|