Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -220,8 +220,6 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
|
|
220 |
trust_remote_code=True
|
221 |
)
|
222 |
|
223 |
-
model.to('cuda')
|
224 |
-
|
225 |
conversation = []
|
226 |
|
227 |
if "file" in message and message["file"]:
|
|
|
220 |
trust_remote_code=True
|
221 |
)
|
222 |
|
|
|
|
|
223 |
conversation = []
|
224 |
|
225 |
if "file" in message and message["file"]:
|