im dumb

914fc5dd · novelailab · cc3440d4 · 914fc5dd
Commit 914fc5dd authored Jul 30, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

main.py main.py +1 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -75,7 +75,7 @@ def generate(request: GenerationRequest):
        process_time = time.perf_counter() - t
        logger.info(f"Request took {process_time:0.3f} seconds")
        print(images)
-        return GenerationRequest(generation=images)
+        return GenerationOutput(generation=images)

    except Exception as e:
        capture_exception(e)