NoMoreCopyrightOrg
/

flux-test5

Model card Files Files and versions

refoundd commited on Mar 24, 2025

Commit

fa6de1f

·

verified ·

1 Parent(s): e6021fb

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -219,7 +219,7 @@ class EndpointHandler:
         data = list(self.collection.find().skip(start_index).limit(limit_count))
         start_time=time.time()
         for document in data:
-                image=load_image(document.get('requestImage', 'https://nomorecopyright.com/default.jpg'))
                 image = image.convert("RGBA")
                 outputs = self.predictor.predict(
                     image,

         data = list(self.collection.find().skip(start_index).limit(limit_count))
         start_time=time.time()
         for document in data:
+                image=load_image(document.get('createdImage', 'https://nomorecopyright.com/default.jpg'))
                 image = image.convert("RGBA")
                 outputs = self.predictor.predict(
                     image,