Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
-title: Age And Emotion Image Classification
-emoji: 🐠
-colorFrom: indigo
-colorTo: pink
-sdk: gradio
-sdk_version: 4.32.0
+title: age_and_emotion_image_classification
 app_file: app.py
-pinned: false
+sdk: gradio
+sdk_version: 4.25.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,30 @@
+# !pip install transformers==4.37.2 gradio==4.25.0
+import gradio as gr
+from transformers import pipeline
+age_classifier = pipeline("image-classification", model="nateraw/vit-age-classifier")
+emotion_classifier = pipeline("image-classification", model="jhoppanne/Image-Emotion-Classification")
+def pred_age_emotion(input_image):
+    if isinstance(input_image,np.ndarray):
+        img = Image.fromarray(input_image)
+        #age classifier
+        age_result = age_classifier(img)
+        age_score = age_result[0].get('score')
+        age_label = age_result[0].get('label')
+        txt1 =''
+        txt1 += f'The Model predict that the person in this image is around {age_label} years old.\n'
+        txt1 += f'with confident score : {age_score*100:.2f}%'
+        #emotion classifier
+        emotion_result = emotion_classifier(img)
+        emotion_score = emotion_result[0].get('score')
+        emotion_label = emotion_result[1].get('label')
+        txt2=''
+        txt2+= f'The Model predict that the emotion of person in this image is {emotion_label}.\n'
+        txt2+= f'with confident score : {emotion_score*100:.2f}% '
+    else:
+        txt1,txt2 =  "sorry, unable to process the image"
+    return txt1, txt2
+    # return f"Data type of uploaded image: {type(img)}"
+def pred_emotion(input_image):
+    return
+iface = gr.Interface(fn=pred_age_emotion, inputs = gr.Image(), outputs = ["text", "text"])
+iface.launch(share=True)

requirements.txt

@@ -0,0 +1,4 @@
+# Python 3.10.14
+gradio==4.25.0
+transformers==4.37.2
+torch