|
|
|
"""Lab10app.ipynb |
|
|
|
Automatically generated by Colaboratory. |
|
|
|
Original file is located at |
|
https://colab.research.google.com/drive/1lrMgwkvs4N_RUAAPjhRcbd1shmVd38Ah |
|
""" |
|
|
|
import streamlit as st |
|
from transformers import pipeline |
|
|
|
classifier = pipeline("image-to-text", model="microsoft/trocr-base-printed") |
|
def main(): |
|
st.title("Yelp review") |
|
|
|
with st.form("text_field"): |
|
text = st.text_area('enter some text:') |
|
|
|
clicked = st.form_submit_button("Submit") |
|
if clicked: |
|
results = classifier([text]) |
|
st.json(results) |
|
|
|
if __name__ == "__main__": |
|
main() |