saffr0n commited on
Commit
f7faa85
1 Parent(s): 696fd28

revert previous change

Browse files
Files changed (1) hide show
  1. app.py +53 -54
app.py CHANGED
@@ -84,64 +84,63 @@ def generate(
84
  outputs.append(text)
85
  yield "".join(outputs)
86
 
87
- chatbot=gr.Chatbot(height=450, label='Gradio ChatInterface')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
88
 
89
  with gr.Blocks(css="style.css") as demo:
90
  gr.Markdown(DESCRIPTION)
91
  gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
92
-
93
- gr.ChatInterface(
94
- fn=generate,
95
- chatbot=chatbot,
96
- fill_height=True,
97
- additional_inputs=[
98
- gr.Textbox(label="System prompt", lines=6),
99
- gr.Slider(
100
- label="Max new tokens",
101
- minimum=1,
102
- maximum=MAX_MAX_NEW_TOKENS,
103
- step=1,
104
- value=DEFAULT_MAX_NEW_TOKENS,
105
- ),
106
- gr.Slider(
107
- label="Temperature",
108
- minimum=0.1,
109
- maximum=4.0,
110
- step=0.1,
111
- value=0.6,
112
- ),
113
- gr.Slider(
114
- label="Top-p (nucleus sampling)",
115
- minimum=0.05,
116
- maximum=1.0,
117
- step=0.05,
118
- value=0.9,
119
- ),
120
- gr.Slider(
121
- label="Top-k",
122
- minimum=1,
123
- maximum=1000,
124
- step=1,
125
- value=50,
126
- ),
127
- gr.Slider(
128
- label="Repetition penalty",
129
- minimum=1.0,
130
- maximum=2.0,
131
- step=0.05,
132
- value=1.2,
133
- ),
134
- ],
135
- stop_btn=None,
136
- examples=[
137
- ["நான் எப்படி வேகமாக தூங்க முடியும்?"],
138
- ["என் முதலாளி மிகவும் கட்டுப்படுத்துகிறார், நான் என்ன செய்ய வேண்டும்?"],
139
- ["திருமணத்திற்கு நான் என்ன அணிய வேண்டும்?"],
140
- ["வரலாற்றில் தெரிந்து கொள்ள வேண்டிய சில முக்கியமான காலங்கள் யாவை?"],
141
- ["நான் பணம் சம்பாதிக்க வேண்டும் ஆனால் வேடிக்கையாக இருக்க வேண்டும் என்றால் நல்ல தொழில் எது?"],
142
- ],
143
- )
144
-
145
  gr.Markdown(LICENSE)
146
 
147
  if __name__ == "__main__":
 
84
  outputs.append(text)
85
  yield "".join(outputs)
86
 
87
+ chat_interface = gr.ChatInterface(
88
+ fn=generate,
89
+ chatbot=chatbot,
90
+ fill_height=True,
91
+ additional_inputs=[
92
+ gr.Textbox(label="System prompt", lines=6),
93
+ gr.Slider(
94
+ label="Max new tokens",
95
+ minimum=1,
96
+ maximum=MAX_MAX_NEW_TOKENS,
97
+ step=1,
98
+ value=DEFAULT_MAX_NEW_TOKENS,
99
+ ),
100
+ gr.Slider(
101
+ label="Temperature",
102
+ minimum=0.1,
103
+ maximum=4.0,
104
+ step=0.1,
105
+ value=0.6,
106
+ ),
107
+ gr.Slider(
108
+ label="Top-p (nucleus sampling)",
109
+ minimum=0.05,
110
+ maximum=1.0,
111
+ step=0.05,
112
+ value=0.9,
113
+ ),
114
+ gr.Slider(
115
+ label="Top-k",
116
+ minimum=1,
117
+ maximum=1000,
118
+ step=1,
119
+ value=50,
120
+ ),
121
+ gr.Slider(
122
+ label="Repetition penalty",
123
+ minimum=1.0,
124
+ maximum=2.0,
125
+ step=0.05,
126
+ value=1.2,
127
+ ),
128
+ ],
129
+ stop_btn=None,
130
+ examples=[
131
+ ["நான் எப்படி வேகமாக தூங்க முடியும்?"],
132
+ ["என் முதலாளி மிகவும் கட்டுப்படுத்துகிறார், நான் என்ன செய்ய வேண்டும்?"],
133
+ ["திருமணத்திற்கு நான் என்ன அணிய வேண்டும்?"],
134
+ ["வரலாற்றில் தெரிந்து கொள்ள வேண்டிய சில முக்கியமான காலங்கள் யாவை?"],
135
+ ["நான் பணம் சம்பாதிக்க வேண்டும் ஆனால் வேடிக்கையாக இருக்க வேண்டும் என்றால் நல்ல தொழில் எது?"],
136
+ ],
137
+ )
138
+
139
 
140
  with gr.Blocks(css="style.css") as demo:
141
  gr.Markdown(DESCRIPTION)
142
  gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
143
+ chat_interface.render()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
144
  gr.Markdown(LICENSE)
145
 
146
  if __name__ == "__main__":