spaces

baseline_utils.py

@@ -6,6 +6,7 @@ import google.generativeai as genai
 from diffusers import AutoPipelineForText2Image
 import torch
 import os
+import spaces
 
 # Utilize the Google Cloud Vision API to recognize text in the
 # input input_images (diary input_images), https://cloud.google.com/vision.
@@ -74,6 +75,7 @@ def analyze_writer_image(image_path, api_key):
 # input_images https://huggingface.co/stabilityai/sdxl-turbo.
 # You can try to output several input_images for a diary entry. Analyze how accurate the results,
 # and think about what could be improved.
+@spaces.GPU
 def generate_comic_book(diary_text, writer_description, num_pages=4):
     pipe = AutoPipelineForText2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
@@ -94,7 +96,7 @@ def generate_comic_book(diary_text, writer_description, num_pages=4):
     #     print("CUDA and MPS not available. Falling back to CPU.")
 
     # Move the model to the selected device
-    pipe = pipe.to('cuda')
+    pipe.to('cuda')
 
     # Create a directory to store the comic book input_images
     os.makedirs("comic_book", exist_ok=True)

requirements.txt

@@ -6,4 +6,5 @@ diffusers
 torch
 gradio
 transformers
-accelerate
+accelerate
+spaces