Update main.py

main.py

@@ -15,6 +15,8 @@ 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'])
+
 #Pydantic object
 class validation(BaseModel):
     prompt: str