bistdude commited on
Commit
fc033f3
1 Parent(s): 24130a4

fixed import

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ if __name__ == "__main__":
29
  """
30
 
31
 
32
- from transformers import pipeline AutoModelForCausalLM, AutoTokenizer
33
  import torch
34
 
35
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
 
29
  """
30
 
31
 
32
+ from transformers import AutoModelForCausalLM, AutoTokenizer
33
  import torch
34
 
35
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"