Spaces:
Sleeping
Sleeping
Updated app with word count and temperature
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from fastai.text.all import *
|
| 2 |
import gradio as gr
|
| 3 |
-
learn = load_learner('
|
| 4 |
-
def predict(text): return learn.predict(text, n_words=
|
| 5 |
-
intr = gr.Interface(predict, "text", "text")
|
| 6 |
-
intr.launch(
|
|
|
|
| 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()
|