Update app.py

app.py

@@ -68,6 +68,7 @@ from .utils import convert_bytes_to_pil
 from diffusers import DiffusionPipeline
 from huggingface_hub.inference_api import InferenceApi
 from huggingface_hub import login
+from datasets import load_dataset
 #from torch import autocast
 #from diffusers import StableDiffusionPipeline
 #from io import BytesIO
@@ -84,6 +85,17 @@ login()
 from huggingface_hub.inference_api import InferenceApi
 inference = InferenceApi(repo_id="bert-base-uncased", token=API_TOKEN)
 
+from datasets import load_dataset
+
+dataset = load_dataset("Fazzie/Teyvat")
+
+from datasets import load_dataset
+
+dataset = load_dataset("Guizmus/AnimeChanStyle")
+
+from datasets import load_dataset
+dataset = load_dataset("Fazzie/Teyvat")
+from datasets import load_dataset
 dataset = load_dataset("Guizmus/AnimeChanStyle")
 
 sys.path.append('src/blip')