ifire commited on
Commit
ac9fb9d
·
verified ·
1 Parent(s): 00b1d4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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(',')]