Update app.py

app.py

@@ -13,7 +13,7 @@ def download_model_NLP():
     model = keras_nlp.models.GPT2CausalLM.from_preset(
     "gpt2_base_en", preprocessor=preprocessor)
     
-    
+    model_clear = model
     #output = "GPT2_keras_deer"
     #id_weights = "106fk9vGF_sHB8St9B2cKYE9jQVNSdKW_"
     #gdown.download(id=id_weights, output=output, quiet=False)
@@ -38,18 +38,19 @@ def download_model_NLP():
     model.load_weights(output)
     #model = tf.keras.models.load_model(output)       
     
-    return model
+    return model_clear, model
 
 
 def get_model():
     return pipeline('text-generation', model='gpt-2') 
 
 def complete_text(start_of_sentence):
-    
+
+    result_clear = model_clear.generate(start_of_sentence, max_length=100)
     result = model.generate(start_of_sentence, max_length=100)
     #result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text']
     return result
 
-model = download_model_NLP()
-iface = gr.Interface(fn=complete_text, inputs=gr.inputs.Textbox(lines=2, placeholder='Start of Sentence Here...'), outputs="text")
+model_clear, model  = download_model_NLP()
+iface = gr.Interface(fn=complete_text, inputs=gr.inputs.Textbox(lines=2, placeholder='Start of Sentence Here...'), outputs=["text", "text"],)
 iface.launch()