Update main.py

main.py

@@ -33,7 +33,7 @@ app.add_middleware(
 # Globals
 executor = ThreadPoolExecutor(max_workers=10)
 llm = None
-upscale_model = None
+aura_sr = None
 client = MongoClient(f"mongodb+srv://hammad:{quote_plus('momimaad@123')}@cluster0.2a9yu.mongodb.net/")
 db = client["Flux"]
 collection = db["chat_histories"]
@@ -58,7 +58,7 @@ async def startup():
         max_tokens=1024,
         api_key="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",
     )
-    upscale_model = AuraSR.from_pretrained("fal/AuraSR-v2")
+    aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
 
 @app.on_event("shutdown")
 def shutdown():
@@ -196,7 +196,7 @@ async def upscale_image(image_url: str, background_tasks: BackgroundTasks):
     def process_image():
         response = requests.get(image_url)
         img = Image.open(BytesIO(response.content))
-        upscaled_image = upscale_model.upscale_4x_overlapped(img)
+        upscaled_image = aura_sr.upscale_4x_overlapped(img)
         filename = f"upscaled_{uuid.uuid4()}.png"
         filepath = save_image_locally(upscaled_image, filename)
         return filepath