m-elio commited on
Commit
2321358
·
verified ·
1 Parent(s): 788905f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -50,8 +50,8 @@ model_name = "m-elio/LLaVA-NDiNO_pt_short_it"
50
  processor = LlavaNextProcessor.from_pretrained(model_name)
51
  model = LlavaNextForConditionalGeneration.from_pretrained(model_name, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, device_map="cuda")
52
 
53
-
54
- image = Image.open("03-Confusing-Pictures.jpg")
55
 
56
  conversation = [
57
  {
@@ -68,3 +68,5 @@ print(processor.decode(output[0][inputs.input_ids.shape[1]:]))
68
  ```
69
 
70
  ## Citation
 
 
 
50
  processor = LlavaNextProcessor.from_pretrained(model_name)
51
  model = LlavaNextForConditionalGeneration.from_pretrained(model_name, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, device_map="cuda")
52
 
53
+ url = "https://www.barnorama.com/wp-content/uploads/2016/12/03-Confusing-Pictures.jpg"
54
+ image = Image.open(requests.get(url, stream=True).raw)
55
 
56
  conversation = [
57
  {
 
68
  ```
69
 
70
  ## Citation
71
+
72
+ TBD