Update app.py

app.py

@@ -1,16 +1,3 @@
 import gradio as gr
 
-from transformers import pipeline
-
-pipe = pipeline("image-classification", model="datasciencemmw/pickonai-best")
-
-def predict(image):
-  return pipe(image)[0]["image_classification"]
-
-iface = gr.Interface(
-  fn=predict, 
-  inputs='image',
-  outputs='label',
-)
-
-iface.launch()
+gr.Interface.load("datascience/pickonai-best").launch()