fffiloni commited on
Commit
3a7df69
1 Parent(s): 02bb63e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import os
3
  import cv2
4
  import numpy as np
 
5
  from moviepy.editor import *
6
  #from share_btn import community_icon_html, loading_icon_html, share_js
7
 
@@ -58,6 +59,7 @@ inference model
58
 
59
  @torch.no_grad()
60
  def xdecoder(image, instruction, *args, **kwargs):
 
61
  image = image.convert("RGB")
62
  with torch.autocast(device_type='cuda', dtype=torch.float16):
63
  return referring_inpainting_gpt3(model_last, image, instruction, *args, **kwargs)
 
2
  import os
3
  import cv2
4
  import numpy as np
5
+ from PIL import Image
6
  from moviepy.editor import *
7
  #from share_btn import community_icon_html, loading_icon_html, share_js
8
 
 
59
 
60
  @torch.no_grad()
61
  def xdecoder(image, instruction, *args, **kwargs):
62
+ image = Image.open(image)
63
  image = image.convert("RGB")
64
  with torch.autocast(device_type='cuda', dtype=torch.float16):
65
  return referring_inpainting_gpt3(model_last, image, instruction, *args, **kwargs)