Update app.py

app.py

@@ -8,19 +8,23 @@ from datasets import load_dataset
 from diffusers import DiffusionPipeline
 from huggingface_hub import login
 import numpy as np
+import spaces
 
+@spaces.GPU
 def guessanImage(model, image):
     imgclassifier  = pipeline("image-classification", model=model)
     if image is not None:  
         description = imgclassifier(image)
     return description
 
+@spaces.GPU
 def guessanAge(model, image):
     imgclassifier  = pipeline("image-classification", model=model)
     if image is not None:  
         description = imgclassifier(image)
     return description    
 
+@spaces.GPU
 def text2speech(model, text, voice):
     print(voice)
     if len(text) > 0:
@@ -34,6 +38,7 @@ def text2speech(model, text, voice):
         audio_data_16bit = (audio_data * 32767).astype(np.int16)
         return speech["sampling_rate"], audio_data_16bit
 
+@spaces.GPU
 def ImageGenFromText(text, model):
     api_key = os.getenv("fluxauth")
     login(token=api_key)