Spaces:
Sleeping
Sleeping
activate cosmos
Browse files- app.py +3 -12
- app_cosmos.py → app_ner.py +12 -3
app.py
CHANGED
@@ -17,17 +17,8 @@ print("Model loading completed")
|
|
17 |
|
18 |
# bu mesaj değiştirilebilir ve chatbotun başlangıç mesajı olarak kullanılabilir
|
19 |
initial_message = [
|
20 |
-
{"role": "system", "content":
|
21 |
-
|
22 |
-
1. Cümle: Cumhurbaşkanı Erdoğan tatile çıktı.
|
23 |
-
Keyword: tatil
|
24 |
-
NER: Cumhurbaşkanı Erdoğan
|
25 |
-
|
26 |
-
2. Cümle: ...
|
27 |
-
Keyword: ...
|
28 |
-
NER: ...
|
29 |
-
"""
|
30 |
-
}
|
31 |
]
|
32 |
|
33 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
@@ -69,7 +60,7 @@ async def ask(request: Request):
|
|
69 |
print("Model process started")
|
70 |
outputs = model.generate(
|
71 |
input_ids,
|
72 |
-
max_new_tokens=
|
73 |
eos_token_id=terminators,
|
74 |
do_sample=True,
|
75 |
temperature=0.6,
|
|
|
17 |
|
18 |
# bu mesaj değiştirilebilir ve chatbotun başlangıç mesajı olarak kullanılabilir
|
19 |
initial_message = [
|
20 |
+
{"role": "system", "content": "Sen bir yapay zeka asistanısın. Kullanıcı sana bir görev verecek. Amacın görevi olabildiğince sadık bir şekilde tamamlamak."}
|
21 |
+
# Görevi yerine getirirken adım adım düşün ve adımlarını gerekçelendir.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
]
|
23 |
|
24 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
60 |
print("Model process started")
|
61 |
outputs = model.generate(
|
62 |
input_ids,
|
63 |
+
max_new_tokens=256,
|
64 |
eos_token_id=terminators,
|
65 |
do_sample=True,
|
66 |
temperature=0.6,
|
app_cosmos.py → app_ner.py
RENAMED
@@ -17,8 +17,17 @@ print("Model loading completed")
|
|
17 |
|
18 |
# bu mesaj değiştirilebilir ve chatbotun başlangıç mesajı olarak kullanılabilir
|
19 |
initial_message = [
|
20 |
-
{"role": "system", "content":
|
21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
]
|
23 |
|
24 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
@@ -60,7 +69,7 @@ async def ask(request: Request):
|
|
60 |
print("Model process started")
|
61 |
outputs = model.generate(
|
62 |
input_ids,
|
63 |
-
max_new_tokens=
|
64 |
eos_token_id=terminators,
|
65 |
do_sample=True,
|
66 |
temperature=0.6,
|
|
|
17 |
|
18 |
# bu mesaj değiştirilebilir ve chatbotun başlangıç mesajı olarak kullanılabilir
|
19 |
initial_message = [
|
20 |
+
{"role": "system", "content":
|
21 |
+
"""Kullanıcı sana bir haber metni verecek. Bu haber metninin önemli kısımlarını özetleyen 5 cümle çıkart. Aynı zamanda bu cümlelerin her birinden bir keyword extract et ve eğer varsa NER ile yer, kişi, tarih gibi alanları extract et. Yoksa karşısını boş bırak. Çıktıların şu formatta olsun:
|
22 |
+
1. Cümle: Cumhurbaşkanı Erdoğan tatile çıktı.
|
23 |
+
Keyword: tatil
|
24 |
+
NER: Cumhurbaşkanı Erdoğan
|
25 |
+
|
26 |
+
2. Cümle: ...
|
27 |
+
Keyword: ...
|
28 |
+
NER: ...
|
29 |
+
"""
|
30 |
+
}
|
31 |
]
|
32 |
|
33 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
69 |
print("Model process started")
|
70 |
outputs = model.generate(
|
71 |
input_ids,
|
72 |
+
max_new_tokens=512,
|
73 |
eos_token_id=terminators,
|
74 |
do_sample=True,
|
75 |
temperature=0.6,
|