Update app.py

app.py

@@ -41,8 +41,8 @@ def generate_prompt(instruction, input=None):
 
 ### Response:"""
 
-model= load_model(name = 's3nh/pythia-410m-70k-steps-self-instruct-polish')
-tokenizer = load_tokenizer(name = 's3nh/pythia-410m-70k-steps-self-instruct-polish')
+# model= load_model(name = 's3nh/pythia-410m-70k-steps-self-instruct-polish')
+# tokenizer = load_tokenizer(name = 's3nh/pythia-410m-70k-steps-self-instruct-polish')
 generation_config = create_generator()
 
 
@@ -71,7 +71,7 @@ def choose_model(name):
 
     return load_model(name), load_tokenizer(name)
 
-model, tokenizer = gr.Interface(choose_model, [gr.inputs.Dropdown(["s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish", "s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish"]), "text"], "text")
+model, tokenizer = gr.Interface(choose_model, [gr.inputs.Dropdown(["s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish", "s3nh/pythia-1.4b-deduped-16k-steps-self-instruct-polish"])])
 
 io = gr.Interface(
     inference,