Update main.py

main.py

@@ -10,7 +10,9 @@ from ctransformers import AutoModelForCausalLM
 from pydantic import BaseModel
 
 config = {"max_seq_len": 4096}
-llm = AutoModelForCausalLM.from_pretrained('MPT-7B-Storywriter-GGML/mpt-7b-storywriter.ggmlv3.q4_0.bin', model_type='mpt')
+llm = AutoModelForCausalLM.from_pretrained('TheBloke/MPT-7B-Storywriter-GGML',
+                                           model_file='mpt-7b-storywriter.ggmlv3.q4_0.bin',
+                                           model_type='mpt')
 app = fastapi.FastAPI()
 app.add_middleware(
     CORSMiddleware,