hw6

app.py

@@ -155,4 +155,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=False, debug=True)
+    demo.launch(share=True, debug=True)

infer.py

@@ -6,6 +6,7 @@ import os
 import spaces
 import gradio as gr
 
+from PIL import Image
 from transformers import pipeline
 from controlnet_aux import MLSDdetector, HEDdetector, NormalBaeDetector, LineartDetector
 from peft import PeftModel, LoraConfig