Minecraft3193092 commited on
Commit
ef31b44
1 Parent(s): f1f9b53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -3,11 +3,17 @@ import os
3
  import sys
4
  import json
5
  import requests
 
 
 
 
 
 
 
 
 
6
 
7
- MODEL = "gpt-3.5-turbo"
8
- API_URL = os.getenv("API_URL")
9
  DISABLED = os.getenv("DISABLED") == 'True'
10
- OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
11
  NUM_THREADS = int(os.getenv("NUM_THREADS"))
12
 
13
  print (NUM_THREADS)
 
3
  import sys
4
  import json
5
  import requests
6
+ import transformers
7
+
8
+ from transformers import pipeline
9
+
10
+ nlp = pipeline("text-generation", model="gpt3", tokenizer="gpt3", device=0) # 指定GPU设备,可选参数
11
+
12
+ # 生成文本
13
+ response = nlp("Translate English to French: I love programming.")
14
+ print(response[0]["generated_text"])
15
 
 
 
16
  DISABLED = os.getenv("DISABLED") == 'True'
 
17
  NUM_THREADS = int(os.getenv("NUM_THREADS"))
18
 
19
  print (NUM_THREADS)