Rename app.py to app

app.py → app

@@ -1,18 +1,17 @@
 import gradio as gr
 from fastai.vision.all import load_learner, PILImage
 
-# Load your learner
+# Import any custom code here (if needed)
+# Example:
+# from your_module import CustomTransform, CustomModel
+
 learn = load_learner('export.pkl')
 
-# Define the prediction function
 def predict(img):
-    img = PILImage.create(img)  # Convert image into PILImage
-    pred, pred_idx, probs = learn.predict(img)  # Make prediction
-    labels = learn.dls.vocab  # Get the labels
+    img = PILImage.create(img)
+    pred, pred_idx, probs = learn.predict(img)
+    labels = learn.dls.vocab
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
-# Define the Gradio interface
 interface = gr.Interface(fn=predict, inputs=gr.Image(), outputs=gr.Label(num_top_classes=3))
-
-# Launch the interface
 interface.launch(share=True)