Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ def main():
|
|
105 |
if st.button("Generate"):
|
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():
|
110 |
st.session_state["photo"]="done"
|
111 |
st.subheader("Summary section, feed your image!")
|
|
|
105 |
if st.button("Generate"):
|
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():
|
110 |
st.session_state["photo"]="done"
|
111 |
st.subheader("Summary section, feed your image!")
|