Update app.py

app.py

@@ -12,8 +12,9 @@ def download_model_NLP():
     sequence_length=128,)
     model = keras_nlp.models.GPT2CausalLM.from_preset(
     "gpt2_base_en", preprocessor=preprocessor)
-    url = "https://drive.google.com/uc?id=1zi3hSBRTP9uwHDVwQMqpGeXb8RBhJfrO"
-    model_path = gdown.download(url, quiet=False)
+
+    id = "1zi3hSBRTP9uwHDVwQMqpGeXb8RBhJfrO"
+    model_path = gdown.download_folder(id=id, quiet=True, use_cookies=False)
     model.load_weights(model_path)
     
     return model
@@ -23,10 +24,11 @@ def get_model():
     return pipeline('text-generation', model='gpt-2') 
 
 def complete_text(start_of_sentence):
-    model = download_model_NLP()
+    
     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")
 iface.launch()