Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,7 +71,7 @@ def king(type ,
|
|
| 71 |
progress=gr.Progress(track_tqdm=True),
|
| 72 |
):
|
| 73 |
if type=="Image Editing" :
|
| 74 |
-
raw_image = Image.open(input_image)
|
| 75 |
inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)
|
| 76 |
out = model.generate(**inputs, min_length=10, max_length=25)
|
| 77 |
caption = processor.decode(out[0], skip_special_tokens=True)
|
|
|
|
| 71 |
progress=gr.Progress(track_tqdm=True),
|
| 72 |
):
|
| 73 |
if type=="Image Editing" :
|
| 74 |
+
raw_image = Image.open(input_image)
|
| 75 |
inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)
|
| 76 |
out = model.generate(**inputs, min_length=10, max_length=25)
|
| 77 |
caption = processor.decode(out[0], skip_special_tokens=True)
|