Skip to content

Commit 21d1cc7

Browse files
authored
fix gens table creation - use _TablesGetter
1 parent 63d3314 commit 21d1cc7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

image_app_simple/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@
1010

1111
# gens database for storing generated image details
1212
tables = database('data/gens.db').t
13-
gens = tables.gens
14-
if not gens in tables:
13+
gens = tables.t.gens
14+
if not gens in tables.t:
1515
gens.create(prompt=str, id=int, folder=str, pk='id')
1616
Generation = gens.dataclass()
1717

@@ -78,4 +78,4 @@ def generate_and_save(prompt, id, folder):
7878
Image.open(requests.get(output[0], stream=True).raw).save(f"{folder}/{id}.png")
7979
return True
8080

81-
if __name__ == '__main__': uvicorn.run("main:app", host='0.0.0.0', port=int(os.getenv("PORT", default=5000)))
81+
if __name__ == '__main__': uvicorn.run("main:app", host='0.0.0.0', port=int(os.getenv("PORT", default=5000)))

0 commit comments

Comments
 (0)