Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+os.system('pip install pyyaml==5.1')
 os.system('pip install transformers==4.25.1')
 os.system('pip install sentencepiece')
 # workaround: install old version of pytorch since detectron2 hasn't released packages for pytorch 1.9 (issue: https://github.com/facebookresearch/detectron2/issues/3158)
@@ -23,8 +24,9 @@ model = LayoutLMv2ForTokenClassification.from_pretrained("amir22010/layoutxlm-xf
 # load image example
 #dataset = load_dataset("ranpox/xfund", 'xfund.ja', split="validation")
 #image = Image.open(dataset[0]["image"][0]).convert("RGB")
-image = Image.open("./ja_val_49.jpg").convert("RGB")
-image.save("document.jpg")
+image1 = Image.open("./ja_val_49.jpg").convert("RGB")
+image1.save("document.jpg")
+
 # define id2label, label2color
 labels = [
     'O',
@@ -93,8 +95,8 @@ css = ".output-image, .input-image {height: 40rem !important; width: 100% !impor
 css = ".image-preview {height: auto !important;}"
 
 iface = gr.Interface(fn=infer, 
-                     inputs=gr.Image(type="pil"), 
-                     outputs=gr.Image(type="pil", label="annotated image"),
+                     inputs=gr.inputs.Image(type="pil"), 
+                     outputs=gr.outputs.Image(type="pil", label="annotated image"),
                      title=title,
                      description=description,
                      article=article,