Update app.py

app.py

@@ -7,8 +7,11 @@ def funksjon(input_img):
     #Må inporterer modellen fra API på huggingface.co 
 
     #imgPred = model.fit(input_img)
-    imgPred = input_img
-    return imgPred
+    learn.load('stage-1')
+    predictions = model.predict(img_array)
+    #imgPred = input_img
+    
+    return predictions
 
 
 demo = gr.Interface(funksjon, gr.Image(), "image")