apjanco commited on
Commit
aaf6a0c
1 Parent(s): 0b75e43

add 7b model

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -33,15 +33,15 @@ def array_to_image_path(image_array):
33
  return full_path
34
 
35
  models = {
36
- "qwen2-vl-fmb": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/qwen2-vl-fmb", torch_dtype="auto").cuda().eval()
37
-
38
  }
39
 
40
  processors = {
41
  "qwen2-vl-fmb": AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
42
  }
43
 
44
- DESCRIPTION = "[Qwen2-VL-2B Spanish Demo](https://huggingface.co/ajanco/qwen2-vl-fmb)"
45
 
46
  kwargs = {}
47
  kwargs['torch_dtype'] = torch.bfloat16
 
33
  return full_path
34
 
35
  models = {
36
+ "qwen2-vl-fmb": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/qwen2-vl-fmb", torch_dtype="auto").cuda().eval(),
37
+ "qwen2-vl-fmb-7B": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/fmb-qwen-vl-7b", torch_dtype="auto").cuda().eval(),
38
  }
39
 
40
  processors = {
41
  "qwen2-vl-fmb": AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
42
  }
43
 
44
+ DESCRIPTION = "[Qwen2-VL 2B or 7B Spanish Demo](https://huggingface.co/ajanco/qwen2-vl-fmb)"
45
 
46
  kwargs = {}
47
  kwargs['torch_dtype'] = torch.bfloat16