vilarin commited on
Commit
36f75a7
1 Parent(s): 7e43c64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -28
app.py CHANGED
@@ -42,9 +42,6 @@ def stream_chat(
42
  history: list,
43
  temperature: float = 0.3,
44
  max_new_tokens: int = 1024,
45
- top_p: float = 1.0,
46
- top_k: int = 20,
47
- penalty: float = 1.2,
48
  ):
49
  print(f'message: {message}')
50
  print(f'history: {history}')
@@ -75,7 +72,7 @@ def stream_chat(
75
 
76
  chatbot = gr.Chatbot(height=600, placeholder=PLACEHOLDER)
77
 
78
- with gr.Blocks(theme="citrus") as demo:
79
  gr.HTML(TITLE)
80
  gr.DuplicateButton(value="Duplicate Space for private use", elem_classes="duplicate-button")
81
  gr.ChatInterface(
@@ -100,30 +97,6 @@ with gr.Blocks(theme="citrus") as demo:
100
  label="Max new tokens",
101
  render=False,
102
  ),
103
- gr.Slider(
104
- minimum=0.0,
105
- maximum=1.0,
106
- step=0.1,
107
- value=1.0,
108
- label="top_p",
109
- render=False,
110
- ),
111
- gr.Slider(
112
- minimum=1,
113
- maximum=20,
114
- step=1,
115
- value=20,
116
- label="top_k",
117
- render=False,
118
- ),
119
- gr.Slider(
120
- minimum=0.0,
121
- maximum=2.0,
122
- step=0.1,
123
- value=1.2,
124
- label="Repetition penalty",
125
- render=False,
126
- ),
127
  ],
128
  examples=[
129
  ["Help me study vocabulary: write a sentence for me to fill in the blank, and I'll try to pick the correct option."],
 
42
  history: list,
43
  temperature: float = 0.3,
44
  max_new_tokens: int = 1024,
 
 
 
45
  ):
46
  print(f'message: {message}')
47
  print(f'history: {history}')
 
72
 
73
  chatbot = gr.Chatbot(height=600, placeholder=PLACEHOLDER)
74
 
75
+ with gr.Blocks(theme="ocean") as demo:
76
  gr.HTML(TITLE)
77
  gr.DuplicateButton(value="Duplicate Space for private use", elem_classes="duplicate-button")
78
  gr.ChatInterface(
 
97
  label="Max new tokens",
98
  render=False,
99
  ),
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
100
  ],
101
  examples=[
102
  ["Help me study vocabulary: write a sentence for me to fill in the blank, and I'll try to pick the correct option."],