Spaces:
Runtime error
Runtime error
PLs work
Browse files
app.py
CHANGED
@@ -181,7 +181,7 @@ context = torch.zeros((1, 1), dtype=torch.long, device=device)
|
|
181 |
def greet(start_character, number_of_tokens):
|
182 |
context[0][0] = encode(start_character)[0]
|
183 |
max_new_tokens = number_of_tokens
|
184 |
-
return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
|
185 |
|
186 |
iface = gr.Interface(fn=greet, inputs=["text", "number"], outputs="text")
|
187 |
-
iface.launch()
|
|
|
181 |
def greet(start_character, number_of_tokens):
|
182 |
context[0][0] = encode(start_character)[0]
|
183 |
max_new_tokens = number_of_tokens
|
184 |
+
return decode(model.generate(context, max_new_tokens=int(max_new_tokens))[0].tolist())
|
185 |
|
186 |
iface = gr.Interface(fn=greet, inputs=["text", "number"], outputs="text")
|
187 |
+
iface.launch(share=True)
|