biffboff commited on
Commit
0301278
·
verified ·
1 Parent(s): 93ce958

Update Gradio components to new API and switch to gpt-neo-2.7B

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -101,7 +101,7 @@ JSON:
101
  inputs = tokenizer.encode(prompt, return_tensors="pt")
102
  outputs = model.generate(
103
  inputs,
104
- max_length=1000,
105
  temperature=0.7,
106
  top_p=0.9,
107
  do_sample=True,
@@ -131,7 +131,7 @@ JSON:"""
131
  inputs = tokenizer.encode(prompt, return_tensors="pt")
132
  outputs = model.generate(
133
  inputs,
134
- max_length=1000,
135
  temperature=0.7,
136
  top_p=0.9,
137
  do_sample=True,
 
101
  inputs = tokenizer.encode(prompt, return_tensors="pt")
102
  outputs = model.generate(
103
  inputs,
104
+ max_length=1000, # You can adjust this value as needed
105
  temperature=0.7,
106
  top_p=0.9,
107
  do_sample=True,
 
131
  inputs = tokenizer.encode(prompt, return_tensors="pt")
132
  outputs = model.generate(
133
  inputs,
134
+ max_length=1000, # You can adjust this value as needed
135
  temperature=0.7,
136
  top_p=0.9,
137
  do_sample=True,