krushna's picture
removed additional .pth
36788dd
raw
history blame contribute delete
317 Bytes
from fastai.text.all import *
import gradio as gr
learn = load_learner('physicscolab_fit.pth')
def predict(text, number_of_words,temperature): return learn.predict(text, n_words=int(number_of_words), temperature=temperature)
intr = gr.Interface(predict, ["text","number", gr.Slider(0.1, 1,0.5)], "text")
intr.launch()