Luis J Camargo commited on
Commit
919ae04
·
1 Parent(s): bef2fad

temperature test

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,7 +88,7 @@ class PaddleOCRModelManager(object):
88
  def download_model():
89
  """Download the fine-tuned Tachiwin model from Hugging Face"""
90
  model_repo = "tachiwin/PaddleOCR-VL-Tachiwin-BF16"
91
- model_repo = "unsloth/PaddleOCR-VL"
92
  model_dir = "./tachiwin_model"
93
 
94
  print(f"Downloading Tachiwin model from {model_repo}...")
@@ -134,7 +134,7 @@ def inference(img):
134
  return "Please upload an image."
135
 
136
  try:
137
- return model_manager.infer(img)
138
 
139
  # # Now extract text from the serialized structure
140
  # extracted_texts = []
 
88
  def download_model():
89
  """Download the fine-tuned Tachiwin model from Hugging Face"""
90
  model_repo = "tachiwin/PaddleOCR-VL-Tachiwin-BF16"
91
+ #model_repo = "unsloth/PaddleOCR-VL"
92
  model_dir = "./tachiwin_model"
93
 
94
  print(f"Downloading Tachiwin model from {model_repo}...")
 
134
  return "Please upload an image."
135
 
136
  try:
137
+ return model_manager.infer(img, min_new_tokens=3, temperature=1.5, min_p=0.1)
138
 
139
  # # Now extract text from the serialized structure
140
  # extracted_texts = []