Spaces:
Sleeping
Sleeping
vtrv.vls
commited on
Commit
·
d1b28b4
1
Parent(s):
9626dd1
API fix
Browse files
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}
|