Spaces:
Sleeping
Sleeping
improved app
Browse files- Pretrained_GPT_med_bot.pth +2 -2
- app.py +1 -1
Pretrained_GPT_med_bot.pth
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a58d2701e932974d8090e3fba87fef031ec7b85a090f0a61d218e2783a876ec4
|
3 |
+
size 128627717
|
app.py
CHANGED
@@ -28,7 +28,7 @@ def main():
|
|
28 |
model.eval()
|
29 |
with torch.inference_mode():
|
30 |
response = model.generate(prompt.unsqueeze(0),
|
31 |
-
max_new_tokens=
|
32 |
answer = decode(response)[in_len:]
|
33 |
|
34 |
pred_time = round(timer() - start, 5)
|
|
|
28 |
model.eval()
|
29 |
with torch.inference_mode():
|
30 |
response = model.generate(prompt.unsqueeze(0),
|
31 |
+
max_new_tokens=100)[0].tolist()
|
32 |
answer = decode(response)[in_len:]
|
33 |
|
34 |
pred_time = round(timer() - start, 5)
|