vtrv.vls commited on
Commit
d1b28b4
·
1 Parent(s): 9626dd1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,8 +40,8 @@ def qwen_gen(content, chat_history):
40
  return '', chat_history
41
 
42
  def model_gen(content, model_type: str, chat_history):
43
- print(model_type)
44
  gen = MODEL_LIB[model_type]
 
45
  return gen(content, chat_history)
46
 
47
  MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}
 
40
  return '', chat_history
41
 
42
  def model_gen(content, model_type: str, chat_history):
 
43
  gen = MODEL_LIB[model_type]
44
+ print(MODEL_LIB[model_type])
45
  return gen(content, chat_history)
46
 
47
  MODEL_LIB = {'RUBASE': giga_gen, 'TINYLLAMA': tiny_gen, 'QWEN2INS1B': qwen_gen}