Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,6 @@ def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_len
|
|
| 27 |
)
|
| 28 |
# Extract the generated text and return it
|
| 29 |
generated_text = outputs[0]['generated_text']
|
| 30 |
-
generator.model.cpu()
|
| 31 |
-
generator.device = torch.device("cpu")
|
| 32 |
return generated_text
|
| 33 |
# Create the Gradio interface
|
| 34 |
iface = gr.Interface(
|
|
|
|
| 27 |
)
|
| 28 |
# Extract the generated text and return it
|
| 29 |
generated_text = outputs[0]['generated_text']
|
|
|
|
|
|
|
| 30 |
return generated_text
|
| 31 |
# Create the Gradio interface
|
| 32 |
iface = gr.Interface(
|