Update app.py

app.py

@@ -13,10 +13,8 @@ import easyocr
 
 # OCR Model
 tokenizer = AutoTokenizer.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, device_map='cpu')
-# model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True)
 model = AutoModel.from_pretrained('RufusRubin777/GOT-OCR2_0_CPU', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
-# model = model.eval().cuda()
-model = model.eval()
+model = model.eval().cpu()
 reader = easyocr.Reader(['hi'])
 
 UPLOAD_FOLDER = "./uploads"
@@ -31,7 +29,7 @@ def image_to_base64(image):
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode()
 
-# OCR Processing of the image uploaded by the user
+
 # @spaces.GPU
 def run_GOT(image,language):
     unique_id = str(uuid.uuid4())
@@ -95,20 +93,7 @@ title_html = """
 <p>Scan Master uses General OCR Theory (GOT), a 580M end-to-end OCR 2.0 model for English optical character recognition and EASYOCR for Hindi optical character recognition. It supports plain text ocr.</p>
 """
 
-acknowledgement_html = """
-<h3>Acknowledgement</h3>
-<a href="https://huggingface.co/ucaslcl/GOT-OCR2_0">[😊 Hugging Face]</a> 
-<a href="https://arxiv.org/abs/2409.01704">[📜 Paper]</a>
-<a href="https://github.com/Ucas-HaoranWei/GOT-OCR2.0/">[🌟 GitHub]</a> 
-"""
-
-aboutme_html = """
-<h3>About Me</h3>
-<p>Name : Satvik Chandrakar</p>
-<a href="https://github.com/Satvik-ai">[🌟 GitHub]</a> """
-
 
-# Scan Master web application developed using Gradio
 with gr.Blocks() as scan_master_web_app:
     gr.HTML(title_html)
     gr.Markdown("""