Spaces:
Runtime error
Runtime error
Stefan Dumitrescu
commited on
Commit
·
d33f166
1
Parent(s):
50f4f41
Update
Browse files- app.py +4 -3
- 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
|