Update app.py

app.py

@@ -1,14 +1,16 @@
 import numpy as np
 import gradio as gr
-
+import torch
+import torchvision
 
 def funksjon(input_img):
     
     #Må inporterer modellen fra API på huggingface.co 
 
     #imgPred = model.fit(input_img)
-    learn.load('stage-1')
-    predictions = model.predict(img_array)
+    model = torch.load('stage-1.pth')
+    
+    predictions = model.predict(input_img)
     #imgPred = input_img
     
     return predictions