Stefan Dumitrescu commited on
Commit
d33f166
·
1 Parent(s): 50f4f41
Files changed (2) hide show
  1. app.py +4 -3
  2. requirements.txt +1 -0
app.py CHANGED
@@ -59,9 +59,9 @@ def infer(model, tokenizer, text, max_length, temperature, top_k, top_p):
59
 
60
  return output_sequences
61
 
62
- model, tokenizer = setModel(model_checkpoint)
63
- output_sequences = infer(model, tokenizer, text_element, max_length, temperature, top_k, top_p)
64
-
65
  for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
66
  print(f"=== GENERATED SEQUENCE {generated_sequence_idx + 1} ===")
67
  generated_sequences = generated_sequence.tolist()
@@ -82,3 +82,4 @@ for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
82
 
83
  st.write(generated_sequences[-1], text_element)
84
 
 
 
59
 
60
  return output_sequences
61
 
62
+ #model, tokenizer = setModel(model_checkpoint)
63
+ #output_sequences = infer(model, tokenizer, text_element, max_length, temperature, top_k, top_p)
64
+ """
65
  for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
66
  print(f"=== GENERATED SEQUENCE {generated_sequence_idx + 1} ===")
67
  generated_sequences = generated_sequence.tolist()
 
82
 
83
  st.write(generated_sequences[-1], text_element)
84
 
85
+ """
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  git+https://github.com/huggingface/transformers
2
  torch
3
  protobuf~=3.19.0
 
1
+ streamlit>=1.12.0
2
  git+https://github.com/huggingface/transformers
3
  torch
4
  protobuf~=3.19.0