NazmulHasanNihal commited on
Commit
82172b4
·
verified ·
1 Parent(s): b8635d1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from openai import OpenAI
2
+
3
+ client = OpenAI(
4
+ base_url = "https://integrate.api.nvidia.com/v1",
5
+ api_key = "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"
6
+ )
7
+
8
+ completion = client.chat.completions.create(
9
+ model="nvidia/nemotron-4-340b-instruct",
10
+ messages=[{"role":"user","content":"Write a limerick about the wonders of GPU computing."}],
11
+ temperature=0.2,
12
+ top_p=0.7,
13
+ max_tokens=1024,
14
+ stream=True
15
+ )
16
+
17
+ for chunk in completion:
18
+ if chunk.choices[0].delta.content is not None:
19
+ print(chunk.choices[0].delta.content, end="")
20
+