Changed to use a Hugging Face Model

app.py

@@ -1,14 +1,21 @@
 import gradio as gr
 from fastai.vision.all import *
+from transformers import AutoImageProcessor, AutoModelForImageClassification
 
 from PIL import Image
+import requests
 
 learn_inf = load_learner("export.pkl")
+processor = AutoImageProcessor.from_pretrained("dima806/facial_emotions_image_detection")
+model = AutoModelForImageClassification.from_pretrained("dima806/facial_emotions_image_detection")
 
 def predict(value) -> str:
-   img = Image.fromarray(value).convert('L')
-   pred,pred_idx,probs = learn_inf.predict(img)
-   return f"{pred} at {probs[pred_idx]}"
+    img = Image.fromarray(value).convert('L')
+    inputs = processor(images=image, return_tensors="pt")
+    outputs = model(**inputs)
+    logits = outputs.logits
+    predicted_class_idx = logits.argmax(-1).item()
+    return model.config.id2label[predicted_class_idx]
 
 with gr.Blocks() as demo:
     with gr.Row():