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",
|