baixintech_zhangyiming_prod commited on
Commit
9c7d2ee
1 Parent(s): 141d444

remove image read

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -1,17 +1,12 @@
1
  import gradio as gr
2
  import gradio.components as grc
3
  import torch
4
- from PIL import Image
5
 
6
  from lavis.models import load_model_and_preprocess
7
  from lavis.processors import load_processor
8
 
9
- raw_image = Image.open("merlion.png").convert("RGB")
10
-
11
  # setup device to use
12
  device = torch.device("cuda") if torch.cuda.is_available() else "cpu"
13
- caption = "merlion in Singapore"
14
-
15
  model, vis_processors, text_processors = load_model_and_preprocess("blip2_image_text_matching", "pretrain", device=device, is_eval=True)
16
 
17
  def predict(raw_image, caption):
 
1
  import gradio as gr
2
  import gradio.components as grc
3
  import torch
 
4
 
5
  from lavis.models import load_model_and_preprocess
6
  from lavis.processors import load_processor
7
 
 
 
8
  # setup device to use
9
  device = torch.device("cuda") if torch.cuda.is_available() else "cpu"
 
 
10
  model, vis_processors, text_processors = load_model_and_preprocess("blip2_image_text_matching", "pretrain", device=device, is_eval=True)
11
 
12
  def predict(raw_image, caption):