Spaces:
Sleeping
Sleeping
Update run.py
Browse files
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,
|
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
|