serhii-korobchenko commited on
Commit
1ded8ac
·
1 Parent(s): 8a5c894

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
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 = get_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