qq li commited on
Commit
b83ace8
·
1 Parent(s): 6289feb
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -1,15 +1,8 @@
1
- from transformers.utils import logging
2
- logging.set_verbosity_error()
3
-
4
- import warnings
5
- warnings.filterwarnings("ignore",
6
- message="Using the model-agnostic default `max_length`")
7
- import os
8
  import gradio as gr
9
  from transformers import pipeline
10
 
11
  pipe = pipeline("image-to-text",
12
- model="./models/Salesforce/blip-image-captioning-base")
13
 
14
  def launch(input):
15
  out = pipe(input)
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
  pipe = pipeline("image-to-text",
5
+ model="Salesforce/blip-image-captioning-base")
6
 
7
  def launch(input):
8
  out = pipe(input)