Kolyadual commited on
Commit
2b10277
·
verified ·
1 Parent(s): 6f1f7b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@
2
  import os
3
  import json
4
  import re
 
5
  import random
6
  from difflib import get_close_matches
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
2
  import os
3
  import json
4
  import re
5
+ import torch
6
  import random
7
  from difflib import get_close_matches
8
  from transformers import AutoTokenizer, AutoModelForCausalLM