prithivMLmods commited on
Commit
ddae8ac
·
verified ·
1 Parent(s): a6c9810

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -76,6 +76,7 @@ coreocr_model = Qwen2VLForConditionalGeneration.from_pretrained(
76
 
77
  # Main Inference Function
78
  @spaces.GPU
 
79
  def model_inference(message, history, use_coreocr):
80
  text = message["text"].strip()
81
  files = message.get("files", [])
 
76
 
77
  # Main Inference Function
78
  @spaces.GPU
79
+ @torch.no_grad()
80
  def model_inference(message, history, use_coreocr):
81
  text = message["text"].strip()
82
  files = message.get("files", [])