Update app.py
Browse files
app.py
CHANGED
|
@@ -103,7 +103,7 @@ def main():
|
|
| 103 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
| 104 |
if st.button("Generate"):
|
| 105 |
tokenizer, model = load_models()
|
| 106 |
-
|
| 107 |
output = model.generate(input_ids, max_length=128)
|
| 108 |
st.success(tokenizer.decode(output[0], skip_special_tokens=True))
|
| 109 |
def change_photo_state():
|
|
|
|
| 103 |
input_ids = tokenizer(message, return_tensors='pt').input_ids
|
| 104 |
if st.button("Generate"):
|
| 105 |
tokenizer, model = load_models()
|
| 106 |
+
st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
|
| 107 |
output = model.generate(input_ids, max_length=128)
|
| 108 |
st.success(tokenizer.decode(output[0], skip_special_tokens=True))
|
| 109 |
def change_photo_state():
|