Update app.py

app.py

@@ -449,14 +449,14 @@ def generate(modelname:str,input_text:str,batch_size:int,inference_steps:int):
     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])
+    pipeline.embedding=torch.tensor([label],device="cpu") #running on free tier
     generate_image=pipeline(batch_size=batch_size,num_inference_steps=inference_steps).images
     return generate_image
 
 
 def switch_pipeline(modelname:str):
     global pipeline
-    pipeline = DiffusionPipeline.from_pretrained(modelname,custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",embedding=torch.float16)
+    pipeline = DiffusionPipeline.from_pretrained(modelname,custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",embedding=torch.int16)
     global current_model
     current_model=modelname
     return f"""