import numpy as np
import gradio as gr
import torch as torch
import torchvision

def funksjon(input_img):
    
    #Må inporterer modellen fra API på huggingface.co 

    #imgPred = model.fit(input_img)
    model = torch.load('stage-1.pth')
    
    predictions = model.predict(input_img)
    #imgPred = input_img
    
    return predictions


demo = gr.Interface(funksjon, gr.Image(), "image")
#demo.launch(share=True)
demo.launch()