witchEverly commited on
Commit
04b8500
·
verified ·
1 Parent(s): df7f37f

Update app_utils.py

Browse files
Files changed (1) hide show
  1. app_utils.py +2 -2
app_utils.py CHANGED
@@ -32,8 +32,8 @@ def init_model(init_model_required):
32
  """
33
  if init_model_required:
34
  try:
35
- processor = AutoProcessor.from_pretrained('./data/pretrained/blip2-opt-2.7b')
36
- blip2_model = Blip2ForConditionalGeneration.from_pretrained('./data/pretrained/blip2-opt-2.7b')
37
  init_model_required = False
38
  return processor, blip2_model, init_model_required
39
  except Exception as e:
 
32
  """
33
  if init_model_required:
34
  try:
35
+ processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
36
+ blip2_model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
37
  init_model_required = False
38
  return processor, blip2_model, init_model_required
39
  except Exception as e: