Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ def upload_model(model_id, tokenizer):
|
|
73 |
|
74 |
return model_repo_name
|
75 |
|
76 |
-
def process(action, system_prompt, user_prompt, base_model_id,
|
77 |
if action == action_1:
|
78 |
result = prompt_model(base_model_id, system_prompt, user_prompt)
|
79 |
elif action == action_2:
|
|
|
73 |
|
74 |
return model_repo_name
|
75 |
|
76 |
+
def process(action, system_prompt, user_prompt, base_model_id, fine_tuned_model_id, dataset):
|
77 |
if action == action_1:
|
78 |
result = prompt_model(base_model_id, system_prompt, user_prompt)
|
79 |
elif action == action_2:
|