Update main.py

main.py

@@ -6,6 +6,7 @@ import subprocess
 import os
 import json
 import uuid
+import html
 
 import torch
 from diffusers import (
@@ -21,6 +22,9 @@ app = FastAPI()
 def generate_image(prompt, model):
     torch.cuda.empty_cache()
 
+    prompt = html.escape(prompt)
+    model = html.escape(model)
+    
     modelArray = model.split(",")
     modelName = modelArray[0]
     modelVersion = modelArray[1]