Update app.py

app.py

@@ -1,11 +1,22 @@
 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"