Update app.py

app.py

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