req

app.py

@@ -1,47 +1,18 @@
 import gradio as gr
-from PIL import Image
 import os
 
 
-def image_mod(image_path):
-    images = []
-    with Image.open(image_path) as im:
-        images.append(im.rotate(45))
-        images.append(im.rotate(90))
-    
-    #files = os.listdir(os.curdir)
-    #print('local files: ',files)
-    #images[0].save("newlion.png")
-    #files = os.listdir(os.curdir)
-    #print('local files: ',files)
+def image_mod(image):
+    return image.rotate(45)
 
-    html = (
-            "<table border=1>"
-            "<tr><td>"
-            "<img  src='file/workdir/lion.jpg' alt='image One'>"
-            "</td>"
-            "<td>"
-            "imapge caption"
-            "</td></tr>"
-            "<tr><td>"
-            "<img  src='file/workdir/cheetah.jpg' alt='image One'>"
-            "</td>"
-            "<td>"
-            "imapge caption"
-            "</td></tr>"
-            "</table>"
-    )
 
-    return html, images
-
-
-demo = gr.Interface(image_mod, "text" , ["html", gr.Gallery()],
-       allow_flagging="never",
-    examples=[
-        os.path.join(os.path.dirname(__file__), "workdir/cheetah.jpg"),
-        os.path.join(os.path.dirname(__file__), "workdir/lion.jpg"),
-        os.path.join(os.path.dirname(__file__), "workdir/logo.png")
+demo = gr.Interface(image_mod, gr.Image(type="pil"), "image",
+    flagging_options=["blurry", "incorrect", "other"], examples=[
+        os.path.join(os.path.dirname(__file__), "images/cheetah1.jpg"),
+        os.path.join(os.path.dirname(__file__), "images/lion.jpg"),
+        os.path.join(os.path.dirname(__file__), "images/logo.png")
         ])
 
 if __name__ == "__main__":
     demo.launch()
+