Update main.py

main.py

@@ -15,11 +15,10 @@ llms = {
 for k in llms.keys():
     AutoModelForCausalLM.from_pretrained(llms[k]['name'], model_file=llms[k]['file'])
 
-llm = AutoModelForCausalLM.from_pretrained(llms["tinnyllama"]['name'], model_file=llms["tinnyllama"]['file'])
-
 #Pydantic object
 class validation(BaseModel):
     prompt: str
+    llm: str
 #Fast API
 app = FastAPI()