revamped code

image_generator.py

@@ -30,14 +30,14 @@
 #     prompt = "A friendly person saying 'How are you?'"
 #     print("Generated Image Path:", generate_image(prompt))
 
-
+import spaces
 from diffusers import StableDiffusionPipeline
 import torch
 
 # Preload the model globally
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
-
+@spaces.GPU
 def generate_image(prompt):
     """Generate an image based on the input prompt."""
     with torch.no_grad():

translate.py

@@ -25,6 +25,7 @@
 #     inputs = tokenizer(text, return_tensors="pt", padding=True)
 #     translated = translation_model.generate(**inputs)
 #     return tokenizer.decode(translated[0], skip_special_tokens=True)
+import spaces
 import torch
 from transformers import MarianMTModel, MarianTokenizer
 
@@ -64,7 +65,7 @@ SUPPORTED_LANGUAGES = {
     "Indonesian": "id",
     "Malay": "ms"
 }
-
+@spaces.GPU
 def translate_text(text, src_lang, tgt_lang="en"):
     """Translate text from the selected source language to English."""
     if src_lang not in SUPPORTED_LANGUAGES: