Spaces:
Build error
Build error
Update app.py
Browse files
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",
|