Update app.py

app.py

@@ -2,13 +2,16 @@ import numpy as np
 import gradio as gr
 import torch as torch
 import torchvision
+import load_model from "./metode"
 
 def funksjon(input_img):
     
     #Må inporterer modellen fra API på huggingface.co 
 
     #imgPred = model.fit(input_img)
-    model = torch.load('stage-1.pth')
+    model = load_model('stage-1.pth')
+    
+    #model = torch.load('stage-1.pth')
     
     predictions = model.predict(input_img)
     #imgPred = input_img