Spaces:

vithacocf
/

ocr

Running on Zero

vithacocf commited on Sep 22

Commit

55c8fcc

verified ·

1 Parent(s): ad04441

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,7 @@ def handle_file(file, prompt, extra_prompt, max_new_tokens, progress=gr.Progress
 # --- 3. Inference Function ---
 @spaces.GPU
 def run_inference(img: Image.Image, prompt: str = "", max_new_tokens: int = 512) -> str:
-     model.to("cuda")
     if img.mode != "RGB":
         img = img.convert("RGB")
     prompt_text = prompt.strip()

 # --- 3. Inference Function ---
 @spaces.GPU
 def run_inference(img: Image.Image, prompt: str = "", max_new_tokens: int = 512) -> str:
+    model.to("cuda")
     if img.mode != "RGB":
         img = img.convert("RGB")
     prompt_text = prompt.strip()