Spaces:

vosatorp
/

style-transfer

Runtime error

style-transfer / app.py

Create app.py

edade5f over 2 years ago

No virus

466 Bytes

	import torch
	import streamlit as st
	import transformers

	st.markdown("hello!")

	model = transformers.AutoModel.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
	tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")

	user_input = st.text_input("Please enter something")

	if len(user_input.split()) > 0:
	outputs = tokenizer(user_input, return_tensors="pt")
	st.markdown(f"{model(**outputs)}")