Update app.py

app.py

@@ -200,15 +200,15 @@ def generate_sgg(image):
             label = obj["id"]
             draw.rectangle(bbox, outline="red", width=3)
             draw.text((bbox[0], bbox[1] - 15), label, fill="red", font=font)
-    except:
-        pass
 
-    return json.dumps(resp) if not isinstance(resp, str) else resp
+        return image, json.dumps(resp)
+    except:
+        return image, resp
 
 gr.Interface(
     fn=generate_sgg,
     inputs=gr.Image(type="pil"),
-    outputs="text",
+    outputs=[gr.Image(type="pil"), gr.Textbox(label="Scene Graph")],
     title="R1-SGG: Compile Scene Graphs with Reinforcement Learning",
     description="Upload an image and generate a structured scene graph in JSON format."
 ).launch()