Update app.py

app.py

@@ -1,22 +1,11 @@
 import gradio as gr
+
+from matplotlib import gridspec
+import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
-from matplotlib import gridspec
-import matplotlib.pyplot as plt
-
-# TensorFlow 버전 충돌 해결을 위해 2.0 버전 이상을 사용
-if tf.__version__.startswith("1."):
-    tf.compat.v1.disable_v2_behavior()
-    tf.compat.v1.enable_eager_execution()
-    tf.compat.v1.InteractiveSession()
-    tf.compat.v1.get_default_graph()
-    tf.compat.v1.disable_eager_execution()
-
-# 모델 및 특성 추출기 로드
-feature_extractor = SegformerFeatureExtractor.from_pretrained("mattmdjaga/segformer_b2_clothes")
-model = TFSegformerForSemanticSegmentation.from_pretrained("mattmdjaga/segformer_b2_clothes")
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "mattmdjaga/segformer_b2_clothes"
@@ -112,9 +101,11 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(),
+                    inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     examples=["person-1.jpg", "person-2.jpg", "person-3.jpg", "person-4.jpg", "person-5.jpg"],
                     allow_flagging='never')
 
+
 demo.launch()
+