fix

app.py

@@ -21,7 +21,7 @@ def error_str(error, title="Error"):
 
 def inference(
     repo_id,
-    pr,
+    discuss_nr,
     prompt,
 ):
 
@@ -35,12 +35,13 @@ def inference(
     dtype = torch.float16 if torch_device == "cuda" else torch.float32
 
     try:
-        pipe = DiffusionPipeline.from_pretrained(repo_id, revision=pr, torch_dtype=dtype)
+        revision = f"refs/pr/{discuss_nr}"
+        pipe = DiffusionPipeline.from_pretrained(repo_id, revision=revision, torch_dtype=dtype)
         pipe.to(torch_device)
 
         return pipe(prompt, generator=generator, num_inference_steps=25).images
     except Exception as e:
-        url = f"https://huggingface.co/{repo_id}/discussions/{pr.split('/')[-1]}"
+        url = f"https://huggingface.co/{repo_id}/discussions/{discuss_nr}"
         message = f"There is a problem with your diffusers weights of the PR: {url}. Error message: \n"
         return None, error_str(message + e)
 
@@ -64,11 +65,11 @@ with gr.Blocks(css="style.css") as demo:
             with gr.Group():
                 repo_id = gr.Textbox(
                     label="Repo id on Hub",
-                    placeholder="Path to model, e.g. CompVis/stable-diffusion-v1-4",
+                    placeholder="Path to model, e.g. CompVis/stable-diffusion-v1-4 for https://huggingface.co/CompVis/stable-diffusion-v1-4",
                 )
-                pr = gr.Textbox(
-                    label="PR branch",
-                    placeholder="PR branch that should be checked, e.g. refs/pr/171",
+                discuss_nr = gr.Textbox(
+                    label="Discussion number",
+                    placeholder="Number of the discussion that should be checked, e.g. 171 for https://huggingface.co/CompVis/stable-diffusion-v1-4/discussions/171",
                 )
                 prompt = gr.Textbox(
                     label="Prompt",
@@ -87,7 +88,7 @@ with gr.Blocks(css="style.css") as demo:
 
     inputs = [
         repo_id,
-        pr,
+        discuss_nr,
         prompt,
     ]
     outputs = [gallery, error_output]