KingNish commited on
Commit
8257673
·
verified ·
1 Parent(s): 7ac8058

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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).convert('RGB')
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)