ToletiSri commited on
Commit
ff1c85b
1 Parent(s): 6b22508

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"]