cagrigungor commited on
Commit
91c0d14
·
verified ·
1 Parent(s): 40b73f2

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +78 -0
app.py ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
+ import gradio as gr
4
+
5
+ MODEL_NAME = "cagrigungor/adrest0"
6
+
7
+ # -------------------------------
8
+ # MODEL ve TOKENIZER YÜKLEME
9
+ # -------------------------------
10
+ device = "cuda" if torch.cuda.is_available() else "cpu"
11
+
12
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
13
+
14
+ # TÜRKÇE KARAKTER PROBLEMİNİ ÇÖZEN SATIR:
15
+ try:
16
+ tokenizer.backend_tokenizer.normalizer = None
17
+ except:
18
+ pass # bazı tokenizer sürümleri için zorunlu değil
19
+
20
+ model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
21
+
22
+
23
+ # -------------------------------
24
+ # ADRES DÜZELTME FONKSİYONU
25
+ # -------------------------------
26
+ def duzelt(adres: str) -> str:
27
+ encoded = tokenizer(
28
+ adres,
29
+ return_tensors="pt",
30
+ truncation=True,
31
+ max_length=64
32
+ ).to(device)
33
+
34
+ with torch.no_grad():
35
+ output = model.generate(
36
+ **encoded,
37
+ max_length=64,
38
+ num_beams=4,
39
+ early_stopping=True
40
+ )
41
+
42
+ text = tokenizer.decode(output[0], skip_special_tokens=True)
43
+ return text
44
+
45
+
46
+ # -------------------------------
47
+ # GRADIO ARAYÜZ
48
+ # -------------------------------
49
+ with gr.Blocks(title="Türkçe Adres Düzeltme - adrest5") as demo:
50
+
51
+ gr.Markdown("""
52
+ # 🏡 Türkçe Adres Düzeltme
53
+ Bu uygulama, bozuk veya eksik yazılmış **Türkçe adresleri** otomatik olarak
54
+ standart ve okunabilir biçime dönüştürür.
55
+ **Model:** `cagrigungor/adrest5`
56
+ """)
57
+
58
+ input_text = gr.Textbox(
59
+ label="Bozuk Adres Girin",
60
+ placeholder="ör: atatrk cad cafega mahs num 18 kadikoy",
61
+ lines=2
62
+ )
63
+
64
+ output_text = gr.Textbox(
65
+ label="Düzeltilmiş Adres",
66
+ lines=2
67
+ )
68
+
69
+ btn = gr.Button("Düzelt")
70
+ btn.click(duzelt, inputs=input_text, outputs=output_text)
71
+
72
+ gr.Markdown("""
73
+ ---
74
+ **Geliştirici:** @cagrigungor
75
+ """)
76
+
77
+
78
+ demo.launch()