Update app.py

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 import torch
 import logging
 from diffusers import DiffusionPipeline
+from models.transformers.transformer_hidream_image import HiDreamImageTransformer2DModel
+from pipelines.hidream_image.pipeline_hidream_image import HiDreamImagePipeline
 import subprocess
 
 from nf4 import *