AFischer1985 commited on
Commit
0d912a8
·
verified ·
1 Parent(s): 3418a2a

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -212,7 +212,7 @@ from datetime import datetime
212
  import os
213
  import re
214
 
215
- def response(message, history,customSysPrompt, genre, hfToken, augmentation):
216
  if((onPrem==False) & (hfToken.startswith("hf_"))): # use HF-hub with custom token if token is provided
217
  from huggingface_hub import InferenceClient
218
  import gradio as gr
 
212
  import os
213
  import re
214
 
215
+ def response(message, history,customSysPrompt, genre, augmentation, hfToken):
216
  if((onPrem==False) & (hfToken.startswith("hf_"))): # use HF-hub with custom token if token is provided
217
  from huggingface_hub import InferenceClient
218
  import gradio as gr