Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ import torch
|
|
9 |
img = "input_data/ocr_input/korean1.jpg"
|
10 |
text = "ννμ΄ μν° κ²λ μλͺ»μΈκ°μ. λ μ°¨κ°μ΄ λμμ λ°λ»ν μ¬μλ°. κ·Έλ₯ μ’μνλ¨ λ§λ μ λλκ°μ. μμ§νκ² λ λ§νκ³ μΆμ΄μ"
|
11 |
model_id = "deepseek-ai/deepseek-llm-7b-chat"
|
|
|
12 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
13 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
|
14 |
|
|
|
9 |
img = "input_data/ocr_input/korean1.jpg"
|
10 |
text = "ννμ΄ μν° κ²λ μλͺ»μΈκ°μ. λ μ°¨κ°μ΄ λμμ λ°λ»ν μ¬μλ°. κ·Έλ₯ μ’μνλ¨ λ§λ μ λλκ°μ. μμ§νκ² λ λ§νκ³ μΆμ΄μ"
|
11 |
model_id = "deepseek-ai/deepseek-llm-7b-chat"
|
12 |
+
|
13 |
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
|
15 |
|