Tonic commited on
Commit
5b6abbe
·
unverified ·
1 Parent(s): 68e5120

space gpu cuda fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -229,7 +229,7 @@ def cuda_error_handler(func):
229
  return f"An unexpected error occurred: {str(e)}", 0, 0
230
  return wrapper
231
 
232
- @spaces.GPU(duration=120)
233
  @cuda_error_handler
234
  def generate_text(image, prompt, max_tokens):
235
  try:
@@ -270,7 +270,7 @@ def generate_text(image, prompt, max_tokens):
270
  traceback.print_exc()
271
  return f"Error: {str(e)}", 0, 0
272
 
273
- @spaces.GPU(duration=60)
274
  @cuda_error_handler
275
  def calculate_similarity(image1, image2):
276
  try:
 
229
  return f"An unexpected error occurred: {str(e)}", 0, 0
230
  return wrapper
231
 
232
+ @spaces.GPU()
233
  @cuda_error_handler
234
  def generate_text(image, prompt, max_tokens):
235
  try:
 
270
  traceback.print_exc()
271
  return f"Error: {str(e)}", 0, 0
272
 
273
+ @spaces.GPU()
274
  @cuda_error_handler
275
  def calculate_similarity(image1, image2):
276
  try: