Add load of dataset in GPU

app.py

@@ -27,6 +27,7 @@ def search_images_from_image(image):
     return fake_gan()
 
 def main():
+    dataset = load_dataset("JLD/unsplash25k-image-embeddings", trust_remote_code=True, split="train").with_format("torch", device="cuda:0")
     text_to_image_iface = gr.Interface(fn=search_images_from_text, inputs="text", outputs="gallery")
     image_to_image_iface = gr.Interface(fn=search_images_from_image, inputs="image", outputs="gallery")
     demo = gr.TabbedInterface([text_to_image_iface, image_to_image_iface], ["Text query", "Image query"])