Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -300,7 +300,7 @@ EXAMPLES = [
|
|
300 |
|
301 |
|
302 |
@spaces.GPU()
|
303 |
-
|
304 |
try:
|
305 |
model = AutoModelForCausalLM.from_pretrained(
|
306 |
MODEL_ID,
|
|
|
300 |
|
301 |
|
302 |
@spaces.GPU()
|
303 |
+
def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
|
304 |
try:
|
305 |
model = AutoModelForCausalLM.from_pretrained(
|
306 |
MODEL_ID,
|