Hugging Face's logo

Spaces:
unography
/
image-captioning-with-longcap
Running on Zero

App Files Community
unography commited on
Commit
26b8d3a
·
verified ·
1 Parent(s): 89c432d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -17,7 +17,6 @@ processor = BlipProcessor.from_pretrained(model_id)
17
  model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)
18
 
19
 
20
- @spaces.CPU
21
  def run(image: PIL.Image.Image) -> str:
22
  inputs = processor(images=image, return_tensors="pt").to(device)
23
  out = model.generate(pixel_values=pixel_values, max_length=300)
 
17
  model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)
18
 
19
 
 
20
  def run(image: PIL.Image.Image) -> str:
21
  inputs = processor(images=image, return_tensors="pt").to(device)
22
  out = model.generate(pixel_values=pixel_values, max_length=300)