Update app.py

app.py

@@ -1,3 +1,8 @@
+# Initialize Google API and model
+import torch
+
+device = torch.device("cpu")  # Force CPU
+
 import base64
 import os
 from huggingface_hub import login
@@ -10,10 +15,7 @@ import google.generativeai as genai
 import gradio as gr  # Add Gradio for UI
 from PIL import Image as PILImage
 
-# Initialize Google API and model
-import torch
 
-device = torch.device("cpu")  # Force CPU
 import os
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=GOOGLE_API_KEY)