Andreas-w commited on
Commit
da1fd41
·
1 Parent(s): 0ef611b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -2,13 +2,16 @@ import numpy as np
2
  import gradio as gr
3
  import torch as torch
4
  import torchvision
 
5
 
6
  def funksjon(input_img):
7
 
8
  #Må inporterer modellen fra API på huggingface.co
9
 
10
  #imgPred = model.fit(input_img)
11
- model = torch.load('stage-1.pth')
 
 
12
 
13
  predictions = model.predict(input_img)
14
  #imgPred = input_img
 
2
  import gradio as gr
3
  import torch as torch
4
  import torchvision
5
+ import load_model from "./metode"
6
 
7
  def funksjon(input_img):
8
 
9
  #Må inporterer modellen fra API på huggingface.co
10
 
11
  #imgPred = model.fit(input_img)
12
+ model = load_model('stage-1.pth')
13
+
14
+ #model = torch.load('stage-1.pth')
15
 
16
  predictions = model.predict(input_img)
17
  #imgPred = input_img