Amitontheweb commited on
Commit
4104b7a
·
verified ·
1 Parent(s): 2c14f20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -24
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
- model_selected.change(
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
  )