Update app.py

app.py

@@ -1,12 +1,14 @@
 # Import necessary libraries
 import gradio as gr
-import transformers_js_py
+import diffusers
+import torch
 
 # Define the function to generate an image from text using the Liberata/illustrious-xl-v1.0 model
 def generate_image(prompt, guidance_scale, num_inference_steps):
     # Load the model from the specified file
-    model = transformers_js_py.StableDiffusionXLPipeline.from_pretrained(
-        "https://huggingface.co/Liberata/illustrious-xl-v1.0/resolve/main/Illustrious-XL-v1.0.safetensors"
+    model = diffusers.StableDiffusionXLPipeline.from_pretrained(
+        "Liberata/illustrious-xl-v1.0",
+        torch_dtype=torch.float16
     )
     # Generate the image from the text prompt with the specified parameters
     image = model(prompt, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]