Update app.py

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
 import random
+import torch  # Assuming your model is compatible with PyTorch for GPU use
 
-# Load the model
-model = gr.load("models/Purz/face-projection")
+# Load the model once at startup (GPU-optimized if available)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model = gr.load("models/Purz/face-projection").to(device)
 
-# Function to generate image
+# Function to generate the image
 def generate_image(text, seed):
     if seed is not None:
         random.seed(seed)
-    return model(text)
+    # Generate image using the model, ensure device compatibility
+    return model(text).to("cpu")  # Move to CPU if Gradio doesn't support GPU display
 
 # Define example inputs
 examples = [
@@ -18,7 +21,7 @@ examples = [
     ["Fantasy dragon", None]
 ]
 
-# Customized Gradio Interface
+# Customized Gradio Interface with faster processing optimizations
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
@@ -28,7 +31,7 @@ interface = gr.Interface(
     outputs=gr.Image(label="Generated Image", elem_id="output-image"),
     examples=examples,
     theme="default",
-    description="The model is currently running on CPU. Performance might vary. Thanks for your patience.",
+    description="The model is currently running on GPU for faster processing. Thanks for your patience.",
     css="""
         #input-text, #seed-slider .input-label, .output-label, .description {
             font-family: 'Arial', sans-serif;