model_name

app.py

@@ -16,7 +16,7 @@ model_dir = "model"
 
 model, tokenizer = load_model_tokenizer_for_generate(model_dir)
 
-def get_completion(prompt, model="dolly-v0-70m"):
+def get_completion(prompt, model_name="dolly-v0-70m"):
 	# Examples from https://www.databricks.com/blog/2023/03/24/hello-dolly-democratizing-magic-chatgpt-open-models.html
 	instructions = [prompt]
 	# set some additional pipeline args