Jyotiyadav commited on
Commit
a9091db
1 Parent(s): 39954b3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from pydantic import BaseModel
4
  import os
5
  auth_token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
6
 
7
- llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.Q4_K_S.gguf",use_auth_token=auth_token)
8
 
9
  #Pydantic object
10
  class validation(BaseModel):
 
4
  import os
5
  auth_token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
6
 
7
+ llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.Q4_K_S.gguf")
8
 
9
  #Pydantic object
10
  class validation(BaseModel):