Update app.py

app.py

@@ -8,6 +8,7 @@ import spaces
 import random
 from dataclasses import dataclass
 from typing import Dict, List, Tuple, Optional
+import json
 
 import gradio as gr
 import numpy as np