Spaces:
Paused
Paused
space gpu cuda fix
Browse files
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(
|
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(
|
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:
|