Spaces:
Sleeping
Sleeping
File size: 515 Bytes
641161a c07154f 641161a c07154f 641161a 496838f 641161a c07154f 496838f 641161a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import torch
import streamlit as st
from transformers import pipeline
title="Text Generator"
classifier = pipeline(task="sentiment-analysis",
model="distilbert-base-uncased-finetuned-sst-2-english")
def main():
st.title(title)
with st.form("questionaire"):
prompt1 = st.text_input("your text")
clicked = st.form_submit_button("Generate image")
if clicked:
result=classifier(prompt1)
st.json(results)
if __name__=='__main__':
main() |