app

app.py

@@ -1,5 +1,15 @@
-import streamlit as st
-from transformers import pipeline
+import gradio as gr
+from fastai import *
 
-pipe = pipeline('image-classification')
+learner_inf = load_learner('baltrishkes.pkl')
 
+def predict(img):
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learner_inf.predict(img)
+    return pred
+    
+
+gr.Interface(fn=predict, 
+             inputs=gr.inputs.Image(shape=(512, 512)),
+             outputs=gr.outputs.Label(),
+             interpretation="default").launch()