rajistics commited on
Commit
a7ac2bb
·
1 Parent(s): a617135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -37,8 +37,6 @@ theme = gr.themes.Monochrome(
37
  ],
38
  )
39
 
40
- inputs = tokenizer.encode(prompt, return_tensors="pt").to(device)
41
- outputs = model.generate(inputs, max_length=100, temperature=0.2)
42
  @spaces.GPU
43
  def generate(
44
  prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, version="StarCoder",
 
37
  ],
38
  )
39
 
 
 
40
  @spaces.GPU
41
  def generate(
42
  prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, version="StarCoder",