Update app.py

app.py

@@ -20,7 +20,7 @@ feature_extractor = None
 
 # Inference function
 @spaces.GPU()
-def run_inference(image, feature_extractor, model, device, valid_ds):
+def run_inference(image, device, valid_ds):
     # Load image from the Gradio input
     input_image = Image.fromarray(image.astype('uint8'), 'RGB')
 
@@ -63,7 +63,7 @@ def run_inference(image, feature_extractor, model, device, valid_ds):
 
 # Create a Gradio interface
 iface = gr.Interface(
-    fn=lambda image: run_inference(image, feature_extractor, model, device, valid_ds),
+    fn=lambda image: run_inference(image, device, valid_ds),
     inputs=gr.inputs.Image(type="numpy"),  # Input is an image
     outputs="text",  # Output is text (predicted class)
     title="Image Classification",