Update app.py

app.py

@@ -3,7 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import List, Optional
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer        #, LlamaTokenizerFast
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig        #, LlamaTokenizerFast
 import re
 import os