Update app.py

app.py

@@ -2,16 +2,11 @@ import torch
 import gradio as gr
 from transformers import pipeline
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
 def predict(image):
     classifier = pipeline(task="image-classification")
     preds = classifier(image)
     return {pred["label"]: round(float(pred["score"]), 4) for pred in preds}
 
-def clear_interface():
-    interface.inputs[0].clear()
-    interface.outputs[0].clear()
 
 description = """
 """
@@ -23,10 +18,7 @@ interface = gr.Interface(
     ],
     outputs=gr.outputs.Label(),
     title="Image Classifier",
-    description=description,
-    buttons=[
-        gr.Button(label="Clear", callback=clear_interface)
-    ]
+    description=description
 )
 
 interface.launch()