Update app.py

app.py

@@ -39,7 +39,11 @@ example_sample = [
 
 from PIL import Image
 def pred_func(image, prompt):
-    out = single_pred_features(image)
+    #out = single_pred_features(image)
+    features ,face_features = single_pred_features(image)
+    req_img = produce_center_crop_image(features ,face_features)
+    out = {}
+    out["spiga_seg"] = req_img
     if type(out) == type({}):
         #return out["spiga_seg"]
         control_image = out["spiga_seg"]
@@ -59,11 +63,10 @@ def pred_func(image, prompt):
         return control_image ,image
 
 
-demo=gr.Interface(fn=pred_func, inputs=['image','text'],
+gr=gr.Interface(fn=pred_func, inputs=['image','text'],
 outputs=[gr.Image(label='output').style(height=512),
 gr.Image(label='output').style(height=512)],
 examples=example_sample if example_sample else None,
 cache_examples = False
 )
-demo.launch(share=False)
-
+gr.launch(share=False)