Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -131,26 +131,10 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
|
|
131 |
|
132 |
#--------ON SELECTING MODEL------------------------
|
133 |
|
134 |
-
def select_model(model_selected):
|
135 |
-
|
136 |
-
global model_name
|
137 |
-
|
138 |
-
if model_selected == "GPT2":
|
139 |
-
model_name = "openai-community/gpt2"
|
140 |
-
#tokenizer = tokenizer_gpt2
|
141 |
-
#model = model_gpt2
|
142 |
-
#print (model_selected + " loaded")
|
143 |
-
|
144 |
-
if model_selected == "Qwen2":
|
145 |
-
model_name = "Qwen/Qwen2-0.5B"
|
146 |
-
#tokenizer = tokenizer_qwen
|
147 |
-
#model = model_qwen
|
148 |
-
|
149 |
# On clicking load button
|
150 |
|
151 |
def load_model (name):
|
152 |
|
153 |
-
global model_name
|
154 |
global tokenizer
|
155 |
global model
|
156 |
|
@@ -399,14 +383,7 @@ with gr.Blocks() as demo:
|
|
399 |
|
400 |
#----------ON SELECTING/CHANGING: RETURN SEEQUENCES/NO OF BEAMS/BEAM GROUPS/TEMPERATURE--------
|
401 |
|
402 |
-
|
403 |
-
fn=select_model, inputs=[model_selected], outputs=[]
|
404 |
-
)
|
405 |
-
|
406 |
-
#num_return_sequences.change(
|
407 |
-
#fn=change_num_return_sequences, inputs=[n_beams,num_return_sequences], outputs=num_return_sequences
|
408 |
-
#)
|
409 |
-
|
410 |
n_beams.change(
|
411 |
fn=popualate_beam_groups, inputs=[n_beams], outputs=[beam_groups,num_return_sequences]
|
412 |
)
|
|
|
131 |
|
132 |
#--------ON SELECTING MODEL------------------------
|
133 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
134 |
# On clicking load button
|
135 |
|
136 |
def load_model (name):
|
137 |
|
|
|
138 |
global tokenizer
|
139 |
global model
|
140 |
|
|
|
383 |
|
384 |
#----------ON SELECTING/CHANGING: RETURN SEEQUENCES/NO OF BEAMS/BEAM GROUPS/TEMPERATURE--------
|
385 |
|
386 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
387 |
n_beams.change(
|
388 |
fn=popualate_beam_groups, inputs=[n_beams], outputs=[beam_groups,num_return_sequences]
|
389 |
)
|