Update main.py

main.py

@@ -15,7 +15,7 @@ llms = {
 for k in llms.keys():
     AutoModelForCausalLM.from_pretrained(llms[k]['name'], model_file=llms[k]['file'])
 
-llm = AutoModelForCausalLM.from_pretrained(llms[0]['name'], model_file=llms[0]['file'])
+llm = AutoModelForCausalLM.from_pretrained(llms["tinnyllama"]['name'], model_file=llms["tinnyllama"]['file'])
 
 #Pydantic object
 class validation(BaseModel):