updated app to work wo stories

app.py

@@ -34,10 +34,10 @@ def get_mask(img_in):
 
         img_square = Image.open(io.BytesIO(base64.b64decode(
             result['square'])))
-        img_stories = Image.open(io.BytesIO(base64.b64decode(
-            result['stories'])))
+        # img_stories = Image.open(io.BytesIO(base64.b64decode(
+        #     result['stories'])))
 
-        return (img_square, img_stories, result['emotion'])
+        return (img_square, result['emotion'])
     else:
         print(f"ERROR: {response.json()['detail']}")
         raise gr.Error(response.json()['detail'])
@@ -69,13 +69,12 @@ with gr.Blocks(title="ARKA + TikTok Demo") as app:
         with gr.Column():
             with gr.Row():
                 output_img = gr.Image(type="pil", label="result_square")
-                output_img1 = gr.Image(type="pil", label="result_stories")
             emotion = gr.Label(label="emotion")
 
-    run_btn.click(get_mask, [input_img], [output_img, output_img1, emotion])
+    run_btn.click(get_mask, [input_img], [output_img, emotion])
 
     with gr.Row():
         gr.HTML(footer)
 
-app.launch(share=False, debug=True, enable_queue=True, show_error=True,
+app.launch(share=True, debug=True, enable_queue=True, show_error=True,
            auth=((os.environ.get("USERNAME"), os.environ.get("PASSWORD"))))