Update app.py

app.py

@@ -6,25 +6,7 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
-from transformers import AutoModelForImageClassification, AutoFeatureExtractor
-import requests
-from io import BytesIO
-
-model_name = "facebook/deit-base-distilled-patch16"
-model = AutoModelForImageClassification.from_pretrained(model_name)
-feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
-
-def classify_image(image):
-    # Load and preprocess the image
-    image = Image.open(BytesIO(image))
-    inputs = feature_extractor(images=image, return_tensors="pt")
-    
-    # Perform image classification
-    with torch.no_grad():
-        outputs = model(**inputs)
-    predicted_class = outputs.logits.argmax().item()
-    
-    return model.config.id2label[predicted_class]
+
 
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
@@ -123,7 +105,7 @@ def sepia(input_img):
 
 
 
-demo = gr.Interface(fn=classify_image,
+demo = gr.Interface(fn=draw_plot,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     title="SWJIN11 TASK",