Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
 from transformers import pipeline
+import torch
 
-pipe = pipeline('sentiment-analysis')
+pipe = pipeline(model="facebook/opt-1.3b", torch_dtype=torch.bfloat16, device_map="auto")
 text = st.text_area("Enter a sentence !")
 
 if text:
-    out = pipe(text)
+    out = pipe(text, do_sample=True, top_p=0.95)
     st.json(out)