diff --git a/image_app_session_credits/main.py b/image_app_session_credits/main.py
index d8f7a66..b5d3a03 100644
--- a/image_app_session_credits/main.py
+++ b/image_app_session_credits/main.py
@@ -1,12 +1,11 @@
-from fastcore.parallel import threaded
-from fasthtml.common import *
-import uuid, os, uvicorn, requests, replicate, stripe
+import requests, uuid, stripe
 from PIL import Image
-from starlette.responses import RedirectResponse
+from fasthtml.common import *
+from replicate import Client
 
 # Replicate setup (for generating images)
 replicate_api_token = os.environ['REPLICATE_API_KEY']
-client = replicate.Client(api_token=replicate_api_token)
+client = Client(api_token=replicate_api_token)
 
 # Stripe (for payments)
 stripe.api_key = os.environ["STRIPE_KEY"]
diff --git a/image_app_session_credits/session.py b/image_app_session_credits/session.py
index f088be0..068d116 100644
--- a/image_app_session_credits/session.py
+++ b/image_app_session_credits/session.py
@@ -1,11 +1,11 @@
-from fastcore.parallel import threaded
-from fasthtml.common import *
-import uuid, os, uvicorn, requests, replicate
+import requests, uuid
 from PIL import Image
+from fasthtml.common import *
+from replicate import Client
 
 # Replicate setup (for generating images)
 replicate_api_token = os.environ['REPLICATE_API_KEY']
-client = replicate.Client(api_token=replicate_api_token)
+client = Client(api_token=replicate_api_token)
 
 # gens database for storing generated image details
 tables = database('gens.db').t
diff --git a/image_app_simple/main.py b/image_app_simple/main.py
index 1ffdc91..ee63013 100644
--- a/image_app_simple/main.py
+++ b/image_app_simple/main.py
@@ -1,17 +1,16 @@
-from fastcore.parallel import threaded
-from fasthtml.common import *
-import uuid, os, uvicorn, requests, replicate
+import requests, uuid
 from PIL import Image
-
+from fasthtml.common import *
+from replicate import Client
 
 # Replicate setup (for generating images)
 replicate_api_token = os.environ['REPLICATE_API_KEY']
-client = replicate.Client(api_token=replicate_api_token)  
+client = Client(api_token=replicate_api_token)
 
 # gens database for storing generated image details
 tables = database('data/gens.db').t
-gens = tables.t.gens
-if not gens in tables.t:
+gens = tables.gens
+if not gens in tables:
     gens.create(prompt=str, id=int, folder=str, pk='id')
 Generation = gens.dataclass()