Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ def inference(input_context, count):
|
|
81 |
out_text = tokenizer_gpt.decode(m.generate(reshaped_tensor, max_new_tokens=count)[0])
|
82 |
return out_text
|
83 |
|
84 |
-
title = "TSAI S22 Assignment: GPT training on LLaMa dataset"
|
85 |
description = "A simple Gradio interface that accepts a context and generates text "
|
86 |
examples = [["Machine Learning","200"],
|
87 |
["Deep Learning","200"]
|
|
|
81 |
out_text = tokenizer_gpt.decode(m.generate(reshaped_tensor, max_new_tokens=count)[0])
|
82 |
return out_text
|
83 |
|
84 |
+
title = "TSAI S22 Assignment: GPT training on LLaMa - redpajama dataset"
|
85 |
description = "A simple Gradio interface that accepts a context and generates text "
|
86 |
examples = [["Machine Learning","200"],
|
87 |
["Deep Learning","200"]
|