Merge branch 'main' of https://huggingface.co/spaces/TIGER-Lab/GenAI-Arena

arena_elo/elo_results_20240811.pkl

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e2f3d61c0154d7f2d19a0bbd3b85b44dbca7a3512a78785379a0494fc8be6201
-size 73808

model/model_manager.py

@@ -72,23 +72,23 @@ class ModelManager:
     
     @spaces.GPU(duration=120)
     def generate_image_ig(self, prompt, model_name):
-        if 'unsafe' not in self.NSFW_filter(prompt):
-            print('The prompt is safe')
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            print(f'The prompt "{prompt}" is not safe')
-            result = ''
+        # if 'unsafe' not in self.NSFW_filter(prompt):
+        print('The prompt is safe')
+        pipe = self.load_model_pipe(model_name)
+        result = pipe(prompt=prompt)
+        # else:
+        #    print(f'The prompt "{prompt}" is not safe')
+        #    result = ''
         return result
 
     def generate_image_ig_api(self, prompt, model_name):
-        if 'unsafe' not in self.NSFW_filter(prompt):
-            print('The prompt is safe')
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            print(f'The prompt "{prompt}" is not safe')
-            result = ''
+        # if 'unsafe' not in self.NSFW_filter(prompt):
+        print('The prompt is safe')
+        pipe = self.load_model_pipe(model_name)
+        result = pipe(prompt=prompt)
+        # else:
+        #    print(f'The prompt "{prompt}" is not safe')
+        #    result = ''
         return result
 
     def generate_image_ig_museum(self, model_name):