Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,8 @@ model_dpo.to("cuda:0")
|
|
17 |
@spaces.GPU
|
18 |
def bot_streaming(message, history, max_tokens, temperature, top_p, top_k, repetition_penalty):
|
19 |
print(max_tokens, temperature, top_p, top_k, repetition_penalty)
|
|
|
|
|
20 |
if message['files']:
|
21 |
image = message['files'][-1]['path']
|
22 |
else:
|
|
|
17 |
@spaces.GPU
|
18 |
def bot_streaming(message, history, max_tokens, temperature, top_p, top_k, repetition_penalty):
|
19 |
print(max_tokens, temperature, top_p, top_k, repetition_penalty)
|
20 |
+
print(message)
|
21 |
+
|
22 |
if message['files']:
|
23 |
image = message['files'][-1]['path']
|
24 |
else:
|