Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,6 +3,9 @@ import os
|
|
| 3 |
os.environ["TORCH_DYNAMO_DISABLE"] = "1"
|
| 4 |
|
| 5 |
import torch
|
|
|
|
|
|
|
|
|
|
| 6 |
import torch._dynamo
|
| 7 |
import gradio as gr
|
| 8 |
import spaces
|
|
@@ -28,14 +31,20 @@ import pytesseract
|
|
| 28 |
from pdf2image import convert_from_path
|
| 29 |
|
| 30 |
# -------------------- μΆκ°: PDF to Markdown λ³ν κ΄λ ¨ import --------------------
|
| 31 |
-
|
| 32 |
-
import
|
| 33 |
-
|
| 34 |
-
import
|
| 35 |
-
import
|
| 36 |
-
import
|
| 37 |
-
import
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
# ---------------------------------------------------------------------------
|
| 40 |
|
| 41 |
# --------------------
|
|
@@ -721,7 +730,10 @@ def stream_chat(
|
|
| 721 |
clear_cuda_memory()
|
| 722 |
|
| 723 |
except Exception as e:
|
| 724 |
-
|
|
|
|
|
|
|
|
|
|
| 725 |
print(f"Stream chat μ€λ₯: {error_message}")
|
| 726 |
clear_cuda_memory()
|
| 727 |
yield "", history + [[message, error_message]]
|
|
|
|
| 3 |
os.environ["TORCH_DYNAMO_DISABLE"] = "1"
|
| 4 |
|
| 5 |
import torch
|
| 6 |
+
# μ±λ₯ μ΅μ νλ₯Ό μν μ€μ (TensorFloat32 μ°μ° νμ±ν)
|
| 7 |
+
torch.set_float32_matmul_precision('high')
|
| 8 |
+
|
| 9 |
import torch._dynamo
|
| 10 |
import gradio as gr
|
| 11 |
import spaces
|
|
|
|
| 31 |
from pdf2image import convert_from_path
|
| 32 |
|
| 33 |
# -------------------- μΆκ°: PDF to Markdown λ³ν κ΄λ ¨ import --------------------
|
| 34 |
+
try:
|
| 35 |
+
import re
|
| 36 |
+
import requests
|
| 37 |
+
from bs4 import BeautifulSoup
|
| 38 |
+
import urllib.request
|
| 39 |
+
import ocrmypdf
|
| 40 |
+
import pytz
|
| 41 |
+
import urllib.parse
|
| 42 |
+
from pypdf import PdfReader
|
| 43 |
+
except ModuleNotFoundError as e:
|
| 44 |
+
raise ModuleNotFoundError(
|
| 45 |
+
"νμ λͺ¨λμ΄ λλ½λμμ΅λλ€. 'beautifulsoup4' ν¨ν€μ§λ₯Ό μ€μΉν΄μ£ΌμΈμ.\n"
|
| 46 |
+
"μ: pip install beautifulsoup4"
|
| 47 |
+
)
|
| 48 |
# ---------------------------------------------------------------------------
|
| 49 |
|
| 50 |
# --------------------
|
|
|
|
| 730 |
clear_cuda_memory()
|
| 731 |
|
| 732 |
except Exception as e:
|
| 733 |
+
# μμΈ λ°μ μ μμΈμ μ 체 μ 보λ₯Ό μΆλ ₯νμ¬ λλ²κΉ
μ λμμ΄ λλλ‘ ν¨
|
| 734 |
+
import traceback
|
| 735 |
+
error_details = traceback.format_exc()
|
| 736 |
+
error_message = f"μ€λ₯κ° λ°μνμ΅λλ€: {str(e)}\n{error_details}"
|
| 737 |
print(f"Stream chat μ€λ₯: {error_message}")
|
| 738 |
clear_cuda_memory()
|
| 739 |
yield "", history + [[message, error_message]]
|