Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from gradio import mix
 
 title = "GPT2"
 description = "Gradio Demo for OpenAI GPT2. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
@@ -7,16 +6,14 @@ description = "Gradio Demo for OpenAI GPT2. To use it, simply add your text, or
 article = "<p style='text-align: center'><a href='https://d4mucfpksywv.cloudfront.net/better-language-models/language_models_are_unsupervised_multitask_learners.pdf' target='_blank'>Language Models are Unsupervised Multitask Learners</a></p>"
 
 examples = [
-    ['Paris is the capital of',"gpt2-medium"]
+    ['Paris is the capital of', "gpt2-medium"]
 ]
 
-io1 = gr.Interface.load("huggingface/distilgpt2")
-
-io2 = gr.Interface.load("huggingface/gpt2-large")
-
-io3 = gr.Interface.load("huggingface/gpt2-medium")
-
-io4 = gr.Interface.load("huggingface/gpt2-xl")
+# Load all models at startup
+io1 = gr.load("huggingface/distilgpt2")
+io2 = gr.load("huggingface/gpt2-large")
+io3 = gr.load("huggingface/gpt2-medium")
+io4 = gr.load("huggingface/gpt2-xl")
 
 def inference(text, model):
     if model == "gpt2-large":
@@ -28,15 +25,23 @@ def inference(text, model):
     else:
         outtext = io1(text)
     return outtext   
-    
-     
 
-gr.Interface(
+# Create the interface
+iface = gr.Interface(
     inference, 
-    [gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["distilgpt2","gpt2-medium","gpt2-large","gpt2-xl"], type="value", default="gpt2-medium", label="model")
-], 
-    gr.outputs.Textbox(label="Output"),
+    [
+        gr.Textbox(label="Input"),
+        gr.Dropdown(
+            choices=["distilgpt2", "gpt2-medium", "gpt2-large", "gpt2-xl"], 
+            value="gpt2-medium", 
+            label="Model"
+        )
+    ], 
+    gr.Textbox(label="Output"),
     examples=examples,
     article=article,
     title=title,
-    description=description).launch(enable_queue=True)
+    description=description
+)
+
+iface.launch(enable_queue=True)