added zero gpu support

app.py

@@ -2,6 +2,7 @@ from diffusers import DiffusionPipeline
 from typing import List, Optional, Tuple, Union
 import torch
 import gradio as gr
+import spaces
 css="""
 #input-panel{
 align-items:center;
@@ -443,13 +444,13 @@ character_mappings_model_wise={
 }
 
 
-
+@spaces.GPU
 def generate(modelname:str,input_text:str,batch_size:int,inference_steps:int):
     batch_size=int(batch_size)
     inference_steps=int(inference_steps)
     print(f"Generating image with label:{character_mappings_model_wise[current_model][input_text]} batch size:{batch_size}")
     label=int(character_mappings_model_wise[current_model][input_text])
-    pipeline.embedding=torch.tensor([label],device="cpu") #running on free tier
+    pipeline.embedding=torch.tensor([label],device="cuda") #testing zero gpu
     generate_image=pipeline(batch_size=batch_size,num_inference_steps=inference_steps).images
     return generate_image
 
@@ -457,6 +458,8 @@ def generate(modelname:str,input_text:str,batch_size:int,inference_steps:int):
 def switch_pipeline(modelname:str):
     global pipeline
     pipeline = DiffusionPipeline.from_pretrained(modelname,custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",embedding=torch.int16)
+    pipeline.to('cuda')
+    
     global current_model
     current_model=modelname
     return f"""