Fix import issue

app.py

@@ -118,7 +118,8 @@ def resize_inputs(image_input, crop_size):
 #     else:
 #         return Image.fromarray(np.zeros([sample_num*256,16*256,3],np.uint8))
 
-def generate(pipe, image_input, target_index):
+def generate(pipe, image_input, azimuth):
+    target_index = round(azimuth % 360 / 22.5)
     output = pipe(conditioning_image=image_input)
     return output[target_index]
 
@@ -278,8 +279,7 @@ def run_demo():
         #                .success(fn=partial(update_guide, _USER_GUIDE2), outputs=[guide_text], queue=False)
 
         # azimuth.render()
-        target_index = round(azimuth % 360 / 22.5)
-        run_btn.click(partial(generate, pipe), inputs=[input_block, target_index], outputs=[output_block], queue=True)\
+        run_btn.click(partial(generate, pipe), inputs=[input_block, azimuth], outputs=[output_block], queue=True)\
                .success(fn=partial(update_guide, _USER_GUIDE3), outputs=[guide_text], queue=False)
 
     demo.queue().launch(share=False, max_threads=80)  # auth=("admin", os.environ['PASSWD'])