Ashish Soni commited on
Commit
b46d27b
·
verified ·
1 Parent(s): cef44ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -6,10 +6,6 @@ from transformers import pipeline
6
  # Initialize Model
7
  get_completion = pipeline("image-to-text",model="Salesforce/blip-image-captioning-base", device=0)
8
 
9
- # def captioner(input):
10
- # output = get_completion(input)
11
- # return output[0]['generated_text']
12
-
13
  @spaces.GPU(duration=120)
14
  def captioner(input: Image.Image) -> str:
15
  """
 
6
  # Initialize Model
7
  get_completion = pipeline("image-to-text",model="Salesforce/blip-image-captioning-base", device=0)
8
 
 
 
 
 
9
  @spaces.GPU(duration=120)
10
  def captioner(input: Image.Image) -> str:
11
  """