update app.py

app.py

@@ -64,7 +64,8 @@ def generate(
     bad_words_ids = [[b] for b in bad_words_ids]
     min_loc_token_id = min(loc_token_ids)
     max_loc_token_id = max(loc_token_ids)
-    image = Image.convert("RGB")
+    image_ori = image
+    image = image.convert("RGB")
     width = image.width
     height = image.height
     image = image.resize((224, 224))
@@ -101,6 +102,7 @@ def generate(
         image_nums=image_nums,
     )
     box = []
+    out_image = image_ori
     for o in outputs[0]:
         if o >= min_loc_token_id and o <= max_loc_token_id:
             box.append(o.item() - min_loc_token_id)