Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ import tempfile
|
|
| 8 |
import os
|
| 9 |
import threading
|
| 10 |
import time
|
|
|
|
| 11 |
|
| 12 |
transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
|
| 13 |
|
|
@@ -24,7 +25,7 @@ def delete_file_after_ten_minutes(filename):
|
|
| 24 |
time.sleep(600) # Wait for 10 minutes
|
| 25 |
os.remove(filename)
|
| 26 |
|
| 27 |
-
|
| 28 |
def predict(text, num_input, num_temp):
|
| 29 |
transformer.eval()
|
| 30 |
labels = [label.strip() for label in text.split(',')]
|
|
|
|
| 8 |
import os
|
| 9 |
import threading
|
| 10 |
import time
|
| 11 |
+
import spaces
|
| 12 |
|
| 13 |
transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
|
| 14 |
|
|
|
|
| 25 |
time.sleep(600) # Wait for 10 minutes
|
| 26 |
os.remove(filename)
|
| 27 |
|
| 28 |
+
@spaces.GPU
|
| 29 |
def predict(text, num_input, num_temp):
|
| 30 |
transformer.eval()
|
| 31 |
labels = [label.strip() for label in text.split(',')]
|