Update CircumSpect/vqa/vqa.py

CircumSpect/vqa/vqa.py

@@ -11,6 +11,7 @@ from io import BytesIO
 from PIL import Image
 import requests
 import torch
+import os
 
 device = setup_device()
 
@@ -26,7 +27,9 @@ def load_image(image_file):
 
 disable_torch_init()
 
-model_path = "models/CRYSTAL-vision"
+model_name = os.environ.get('VLM_MODEL')
+
+model_path = "models/CRYSTAL-vision" if model_name != None else model_name
 model_base = None
 conv_mode = None
 temperature = 0.2