Spaces:

Junjie96
/

UniPortrait

Running

App Files Files Community

Junjie96 commited on 27 days ago

Commit

6475329

•

1 Parent(s): 4d77ce5

add nsfw safety checker

Files changed (1) hide show

src/generation.py +1 -1

src/generation.py CHANGED Viewed

@@ -15,7 +15,7 @@ def call_generation(data):
     model_id = os.getenv("MODEL_ID")
     url_query = os.getenv("URL_QUERY")
-    batch_size = 2
     repeat_times = 1
     headers = {

     model_id = os.getenv("MODEL_ID")
     url_query = os.getenv("URL_QUERY")
+    batch_size = 4
     repeat_times = 1
     headers = {