mouaddb commited on
Commit
8c78b64
·
1 Parent(s): 95f4cfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -2,6 +2,15 @@ import gradio as gr
2
  from transformers import AutoProcessor, AutoTokenizer, AutoImageProcessor, AutoModelForCausalLM, BlipForConditionalGeneration, VisionEncoderDecoderModel
3
  import torch
4
 
 
 
 
 
 
 
 
 
 
5
  git_processor_base = AutoProcessor.from_pretrained("microsoft/git-base-coco")
6
  git_model_base = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
7
 
 
2
  from transformers import AutoProcessor, AutoTokenizer, AutoImageProcessor, AutoModelForCausalLM, BlipForConditionalGeneration, VisionEncoderDecoderModel
3
  import torch
4
 
5
+ torch.hub.download_url_to_file('./test-1.jpg', 'test-1.jpg')
6
+ torch.hub.download_url_to_file('./test-2.jpg', 'test-2.jpg')
7
+ torch.hub.download_url_to_file('./test-3.jpg', 'test-3.jpg')
8
+ torch.hub.download_url_to_file('./test-4.jpg', 'test-4.jpg')
9
+ torch.hub.download_url_to_file('./test-5.jpg', 'test-5.jpg')
10
+ torch.hub.download_url_to_file('./test-6.jpg', 'test-6.jpg')
11
+ torch.hub.download_url_to_file('./test-7.jpg', 'test-7.jpg')
12
+
13
+
14
  git_processor_base = AutoProcessor.from_pretrained("microsoft/git-base-coco")
15
  git_model_base = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
16