Update app.py

app.py

@@ -1,17 +1,10 @@
 import streamlit as st
-import torch
-from transformers import AutoTokenizer, GPTNeoForCausalLM
-from PIL import Image
-import io
+from transformer import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("runwayml/stable-diffusion-v1-5")
-model = GPTNeoForCausalLM.from_pretrained("runwayml/stable-diffusion-v1-5")
+pipe = pipeline('sentiment-analysis')
 
-st.title("Scribble Drawing Generator")
+text =st.text area('enter some text!')
 
-prompt = st.text_input("Enter a prompt:")
-if prompt:
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(inputs['input_ids'], max_length=256, do_sample=True)
-    image = Image.open(io.BytesIO(outputs[0].cpu().numpy()))
-    st.image(image, caption="Generated Drawing")
+if text:
+    out = pipe(text)
+    st.json(out)