Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 from fastai.vision.all import *
 
+# Define the GrayscaleTransform class
+class GrayscaleTransform(Transform):
+    def encodes(self, img: PILImage):
+        return img.convert("L")
+
 # Load the exported model
 learn = load_learner('clocker.pkl')
 
 def classify_image(img):
-    # Make a prediction
-    pred, idx, probs = learn.predict(img)
+    pred, _, probs = learn.predict(img)
     return {
         "average woman": float(probs[0]),
         "transgender woman": float(probs[1])
@@ -17,9 +21,9 @@ iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(),
     outputs=gr.Label(num_top_classes=2),
-    title="Woman Type Classifier",
-    description="Upload an image to classify whether it's an average woman or a transgender woman.",
-    examples=["average_woman.jpg"]
+    title="Transfem Clocker AI",
+    description="Upload an image of a woman and this will guess if she is trans.",
+   
 )
 
 # Launch the interface