Update main.py

main.py

@@ -1,6 +1,6 @@
 import json
 import markdown
-from typing import List, Dict, Any, Generator
+from typing import Callable, List, Dict, Any, Generator
 from functools import partial
 
 import fastapi
@@ -14,8 +14,6 @@ from anyio.to_thread import run_sync
 from ctransformers import AutoModelForCausalLM
 from pydantic import BaseModel
 
-
-
 llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardCoder-15B-1.0-GGML",
                                            model_file="WizardCoder-15B-1.0.ggmlv3.q4_0.bin",
                                            model_type="starcoder")