Matthev00 commited on
Commit
914b0b5
1 Parent(s): 4d2039a

improved app

Browse files
Files changed (2) hide show
  1. Pretrained_GPT_med_bot.pth +2 -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:ce9906732c175852c397380acc7298cd42fe5019b3744faef55d04d3536096eb
3
- size 128625029
 
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=200)[0].tolist()
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)