fix 01

main.py

@@ -2,7 +2,15 @@ import os
 import torch
 from fastapi import FastAPI
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import uvicorn
+from fastapi.middleware.cors import CORSMiddleware
 app = FastAPI(docs_url="/")
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 tokenizer = AutoTokenizer.from_pretrained("succinctly/text2image-prompt-generator")
 model = AutoModelForCausalLM.from_pretrained("succinctly/text2image-prompt-generator")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")