sci-m-wang commited on
Commit
b41be8b
verified
1 Parent(s): d566549

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,8 +3,10 @@ from random import choices, randint
3
  # from openai import OpenAI
4
  import os
5
  os.system("pip install transformers")
 
6
  import json
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
8
 
9
  """
10
  璋佹槸鍗у簳娓告垙
 
3
  # from openai import OpenAI
4
  import os
5
  os.system("pip install transformers")
6
+ os.system("pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu")
7
  import json
8
  from transformers import AutoModelForCausalLM, AutoTokenizer
9
+ import torch
10
 
11
  """
12
  璋佹槸鍗у簳娓告垙