From 3a47bd18e858e249b12fc0925371318df72275d8 Mon Sep 17 00:00:00 2001
From: Lian Junhong <69239617+T-Atlas@users.noreply.github.com>
Date: Fri, 17 Mar 2023 10:30:27 +0800
Subject: [PATCH] Update generate.py

Adapting to the input function, a text box for inputting content has been added.
---
 generate.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/generate.py b/generate.py
index f6c78be..a880d22 100644
--- a/generate.py
+++ b/generate.py
@@ -45,13 +45,13 @@ model.eval()
 
 
 def evaluate(
-    instruction,
-    temperature=0.1,
-    top_p=0.75,
-    top_k=40,
-    num_beams=4,
-    input=None,
-    **kwargs,
+        instruction,
+        input=None,
+        temperature=0.1,
+        top_p=0.75,
+        top_k=40,
+        num_beams=4,
+        **kwargs,
 ):
     prompt = generate_prompt(instruction, input)
     inputs = tokenizer(prompt, return_tensors="pt")
@@ -82,6 +82,9 @@ gr.Interface(
         gr.components.Textbox(
             lines=2, label="Instruction", placeholder="Tell me about alpacas."
         ),
+        gr.components.Textbox(
+            lines=2, label="Input", placeholder="none"
+        ),
         gr.components.Slider(minimum=0, maximum=1, value=0.1, label="Temperature"),
         gr.components.Slider(minimum=0, maximum=1, value=0.75, label="Top p"),
         gr.components.Slider(minimum=0, maximum=100, step=1, value=40, label="Top k"),