bstraehle commited on
Commit
3a9c101
1 Parent(s): df44c11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, dataset, fine_tuned_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: