mrm8488 commited on
Commit
f571a7e
1 Parent(s): a0d0b4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model_id = "mrm8488/idefics-9b-ft-describe-diffusion-bf16"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
  model = IdeficsForVisionText2Text.from_pretrained(model_id, torch_dtype=torch.bfloat16)
10
- processor = AutoProcessor.from_pretrained(config.base_model_name_or_path)
11
 
12
  def predict(prompt, image_url, max_length):
13
  image = processor.image_processor.fetch_images(image_url)
 
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
  model = IdeficsForVisionText2Text.from_pretrained(model_id, torch_dtype=torch.bfloat16)
10
+ processor = AutoProcessor.from_pretrained(model_id)
11
 
12
  def predict(prompt, image_url, max_length):
13
  image = processor.image_processor.fetch_images(image_url)