Ajuste doc converter

app.py

@@ -10,14 +10,9 @@ import config
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
 
-# @spaces.GPU
-# def greet(n):
-#     print(zero.device) # <-- 'cuda:0' 🤗
-#     return f"Hello {zero + n} Tensor"
-
 MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
 
-name_model = "Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8" #"TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+name_model = "Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8"
 config.local_model_path = snapshot_download(
     repo_id=name_model,
     cache_dir="/root/.cache/huggingface",
@@ -37,14 +32,11 @@ def process_file(file):
     if file_size > MAX_FILE_SIZE:
         return f"O arquivo excede o limite. Por favor, realize o upload de um arquivo que contenha no máximo {MAX_FILE_SIZE/1024/1024:.1f}MB."
 
-    file_obj, type_file = fix_type(file)
-    texto_extraido = None
-
-    if type_file in EXTENSIONS_FILES:
-        texto_extraido = extract_content_in_pdf(file_obj)
-    elif type_file in EXTENSIONS_IMG_FILES:
-        texto_extraido = doc_converter(file)
-    return texto_extraido or "Não foi possível extrair texto."
+    texto_extraido = doc_converter(file)
+    if texto_extraido is None:
+        return "Não foi possível extrair o texto."
+    
+    return texto_extraido
 
 @spaces.GPU
 def ask_question(texto_extraido, question):

utils.py

@@ -21,9 +21,13 @@ def fix_type(file_upload):
 
 @spaces.GPU
 def doc_converter(file_path):
-    converter = DocumentConverter()
-    result = converter.convert(file_path)
-    return result.document.export_to_markdown()
+    try:
+        converter = DocumentConverter()
+        result = converter.convert(file_path)
+        return result.document.export_to_markdown()
+    except Exception as ex:
+        print(ex)
+        return None
 
 # Resize image while maintaining aspect ratio
 def resize_image(image, max_size):