Update app.py

app.py

@@ -12,7 +12,7 @@ import numpy as np
 import gradio as gr
 from transformers import AutoModel, AutoTokenizer
 from pytorch_pretrained_biggan import BigGAN
-from huggingface_task.autoencoder import AutoEncoder
+from huggingface_task.models import AutoEncoder
 from huggingface_task.run_model import generate_image_from_text
 
 biggan_id = 'biggan-deep-128'