Spaces:
Sleeping
Sleeping
File size: 541 Bytes
19694c1 b712878 19694c1 b712878 41beb64 19694c1 86126f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from fastai.vision.all import *
import gradio as gr
import torch
import numpy as np
from torchvision import transforms
def predict(text):
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = torch.jit.load("model.pth")
model = model.cpu()
model.eval()
model.to(device)
with torch.no_grad():
outputs = model(text)
outputs = torch.argmax(outputs,1)
return(outputs)
# Creamos la interfaz y la lanzamos.
gr.Interface(fn=predict, inputs="text", outputs="text").launch(); |