krushna commited on
Commit
a8f7bfe
·
1 Parent(s): bb4a0bf

Updated app with word count and temperature

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from fastai.text.all import *
2
  import gradio as gr
3
- learn = load_learner('imdb10epoch.pth')
4
- def predict(text): return learn.predict(text, n_words=30)
5
- intr = gr.Interface(predict, "text", "text")
6
- intr.launch(inline =False)
 
1
  from fastai.text.all import *
2
  import gradio as gr
3
+ learn = load_learner('imdb10epochexport.pth')
4
+ def predict(text, number_of_words,temperature): return learn.predict(text, n_words=int(number_of_words), temperature=temperature)
5
+ intr = gr.Interface(predict, ["text","number", gr.Slider(0.1, 1,0.5)], "text")
6
+ intr.launch()