[MNT] Update webui.py

lh11-bb · May 22, 2023 · 5807e73 · 5807e73
1 parent 2fbc20a
commit 5807e73
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,4 +15,5 @@ outputs/*
 data/*
 !data/.gitkeep
 wandb/
+flagged/
 .DS_Store
diff --git a/webui.py b/webui.py
@@ -95,7 +95,7 @@ def main(
 
     def evaluate(
         instruction,
-        input=None,
+        # input=None,
         temperature=0.1,
         top_p=0.75,
         top_k=40,
@@ -104,6 +104,7 @@ def evaluate(
         stream_output=False,
         **kwargs,
     ):
+        input=None
         prompt = prompter.generate_prompt(instruction, input)
         inputs = tokenizer(prompt, return_tensors="pt")
         input_ids = inputs["input_ids"].to(device)
@@ -175,9 +176,9 @@ def generate_with_streaming(**kwargs):
             gr.components.Textbox(
                 lines=2,
                 label="Instruction",
-                placeholder="Tell me about alpacas.",
+                placeholder="此处输入法律相关问题",
             ),
-            gr.components.Textbox(lines=2, label="Input", placeholder="none"),
+            # gr.components.Textbox(lines=2, label="Input", placeholder="none"),
             gr.components.Slider(
                 minimum=0, maximum=1, value=0.1, label="Temperature"
             ),
@@ -197,7 +198,7 @@ def generate_with_streaming(**kwargs):
         ],
         outputs=[
             gr.inputs.Textbox(
-                lines=5,
+                lines=8,
                 label="Output",
             )
         ],