Spaces:
Running
on
Zero
Running
on
Zero
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]]
|