anirudh-sub commited on
Commit
7379b19
·
1 Parent(s): a83fa2d

Slowly debugging app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -12,6 +12,16 @@ pipeline = transformers.pipeline(
12
  device_map="auto",
13
  )
14
 
 
 
 
 
 
 
 
 
 
 
15
  def greet(name):
16
  return "Hello " + name + "!"
17
 
 
12
  device_map="auto",
13
  )
14
 
15
+ sequences = pipeline(
16
+ "What is Extra-T in debate?",
17
+ do_sample=True,
18
+ top_k=10,
19
+ num_return_sequences=1,
20
+ eos_token_id=tokenizer.eos_token_id,
21
+ max_length=100,
22
+ )
23
+
24
+
25
  def greet(name):
26
  return "Hello " + name + "!"
27