Spaces:
Runtime error
Runtime error
Commit
·
1ded8ac
1
Parent(s):
8a5c894
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,25 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
|
4 |
def get_model():
|
5 |
return pipeline('text-generation', model='gpt-2')
|
6 |
|
7 |
def complete_text(start_of_sentence):
|
8 |
-
model =
|
9 |
result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text']
|
10 |
return result
|
11 |
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
+
import keras_nlp
|
4 |
+
import tensorflow as tf
|
5 |
+
import gdown
|
6 |
+
from tensorflow import keras
|
7 |
+
import time
|
8 |
+
|
9 |
+
def download_model_NLP():
|
10 |
+
preprocessor = keras_nlp.models.GPT2CausalLMPreprocessor.from_preset(
|
11 |
+
"gpt2_base_en",
|
12 |
+
sequence_length=128,)
|
13 |
+
model = keras_nlp.models.GPT2CausalLM.from_preset(
|
14 |
+
"gpt2_base_en", preprocessor=preprocessor)
|
15 |
+
return model
|
16 |
+
|
17 |
|
18 |
def get_model():
|
19 |
return pipeline('text-generation', model='gpt-2')
|
20 |
|
21 |
def complete_text(start_of_sentence):
|
22 |
+
model = download_model_NLP()
|
23 |
result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text']
|
24 |
return result
|
25 |
|