Spaces:
Running
on
Zero
Running
on
Zero
fix zerogpu error
Browse files
app.py
CHANGED
|
@@ -85,6 +85,7 @@ def _gc():
|
|
| 85 |
|
| 86 |
|
| 87 |
def _launch_demo(args, model, processor):
|
|
|
|
| 88 |
def call_local_model(model, processor, messages):
|
| 89 |
print(messages)
|
| 90 |
messages = [messages]
|
|
|
|
| 85 |
|
| 86 |
|
| 87 |
def _launch_demo(args, model, processor):
|
| 88 |
+
@spaces.GPU(duration=200)
|
| 89 |
def call_local_model(model, processor, messages):
|
| 90 |
print(messages)
|
| 91 |
messages = [messages]
|