Update app.py

app.py

@@ -6,6 +6,8 @@ import torch
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import spaces
+from comfy import model_management
+from nodes import NODE_CLASS_MAPPINGS
 
 hf_hub_download(
     repo_id="Comfy-Org/stable-diffusion-v1-5-archive",