11

app.py

@@ -79,18 +79,18 @@ def FlowEditRun(
 
     ):
 
-    # if oauth_token is None:
-    #     raise gr.Error("You must be logged in to use Stable Diffusion 3.0 and FLUX.1 models.")
-    # if model_type == 'SD3':
-    #     try:
-    #         huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(SD3STRING, 'sd3_medium.safetensors'),
-    #                                                 token=oauth_token.token)
-    #         print('Has Access')
-    #     # except huggingface_hub.utils._errors.GatedRepoError:
-    #     except huggingface_hub.errors.GatedRepoError:
-    #         raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
-    #                         "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>"
-    #                         "model page</a> to get access.")
+    if oauth_token is None:
+        raise gr.Error("You must be logged in to use Stable Diffusion 3.0 and FLUX.1 models.")
+    if model_type == 'SD3':
+        try:
+            huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(SD3STRING, 'sd3_medium.safetensors'),
+                                                    token=oauth_token.token)
+            print('Has Access')
+        # except huggingface_hub.utils._errors.GatedRepoError:
+        except huggingface_hub.errors.GatedRepoError:
+            raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
+                            "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>"
+                            "model page</a> to get access.")
     # elif model_type == 'FLUX':
     #     try:
     #         huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(FLUXSTRING, 'flux1-dev.safetensors'),