Update app.py

app.py

@@ -1,25 +1,12 @@
-from huggingface_hub import from_pretrained_keras
 import gradio as gr
-import tensorflow as tf
-
-model = from_pretrained_keras("araeynn/validation_model")
-
+from huggingface_hub import AsyncInferenceClient
+client = AsyncInferenceClient()
 
 def image_classifier(inp):
     class_names = ["0", "1"]
     inp.save("why.png")
     sunflower_path = "why.png"
-    img = tf.keras.utils.load_img(sunflower_path, target_size=(256, 256))
-    img_array = tf.keras.utils.img_to_array(img)
-    img_array = tf.expand_dims(img_array, 0)  # Create a batch
-
-    predictions = model.predict(img_array)
-    score = tf.nn.softmax(predictions)
-    r = {}
-    print(score)
-    for s in score:
-        print(s)
-    r = {"0":score, "1":1 - score}
+    r = await client.zero_shot_image_classification("why.png", labels=["mouth", "other"])
     return r