Spaces:
Sleeping
Sleeping
Roaoch commited on
Commit ·
549722f
1
Parent(s): 82ad620
Answer problem
Browse files- src/cyberclaasic.py +1 -1
src/cyberclaasic.py
CHANGED
|
@@ -46,7 +46,7 @@ class CyberClassic(torch.nn.Module):
|
|
| 46 |
return decoded[index]
|
| 47 |
|
| 48 |
def answer(self, promt: str) -> str:
|
| 49 |
-
promt = promt + '
|
| 50 |
length = len(promt)
|
| 51 |
|
| 52 |
promt_tokens = self.tokenizer(promt, return_tensors='pt')
|
|
|
|
| 46 |
return decoded[index]
|
| 47 |
|
| 48 |
def answer(self, promt: str) -> str:
|
| 49 |
+
promt = promt + '. '
|
| 50 |
length = len(promt)
|
| 51 |
|
| 52 |
promt_tokens = self.tokenizer(promt, return_tensors='pt')
|