krushna's picture
inline false
8c5d679
raw
history blame contribute delete
331 Bytes
from fastai.text.all import *
import gradio as gr
learn = load_learner('imdb10epochcolab.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(inline = False)