Spaces:

prithivMLmods
/

DocScope-R1

Running on Zero

prithivMLmods commited on 20 days ago

Commit

47d7775

verified ·

1 Parent(s): 792fff9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,7 @@ def calc_timeout_generic(*args, **kwargs):
         return 60
-@spaces.GPU(duration=calc_timeout_generic)
 def generate_image(model_name, text, image, max_new_tokens=1024, temperature=0.6, top_p=0.9, top_k=50, repetition_penalty=1.2, gpu_timeout=60):
     try:
         if not model_name or model_name not in MODEL_MAP:

         return 60
+@spaces.GPU(duration=calc_timeout_generic, size="xlarge")
 def generate_image(model_name, text, image, max_new_tokens=1024, temperature=0.6, top_p=0.9, top_k=50, repetition_penalty=1.2, gpu_timeout=60):
     try:
         if not model_name or model_name not in MODEL_MAP: