humblemikey commited on
Commit
b129969
1 Parent(s): 5763a51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def process_image(image):
44
 
45
  prompt = "<MORE_DETAILED_CAPTION>"
46
 
47
- inputs = processor(text=prompt, images=image, return_tensors="pt")#.to("cuda")
48
  generated_ids = model.generate(
49
  input_ids=inputs["input_ids"],
50
  pixel_values=inputs["pixel_values"],
 
44
 
45
  prompt = "<MORE_DETAILED_CAPTION>"
46
 
47
+ inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
48
  generated_ids = model.generate(
49
  input_ids=inputs["input_ids"],
50
  pixel_values=inputs["pixel_values"],