use degen-tweet-generator-v1

app.py

@@ -1,13 +1,12 @@
 import streamlit as st
 from transformers import pipeline
-from datasets import load_dataset
 
-load_dataset("kaifahmad/Hate-Speech-Tweets")
-load_dataset("m-newhauser/distilbert-political-tweets")
-# pipe = pipeline("text-generation", model="cognitivecomputations/dolphin-2.7-mixtral-8x7b")
-pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B", device=1)
+pipe = pipeline("text-generation", model="yubi12/autotrain-crypto-degen-tweet-generator-v1", device=1)
 text = st.text_input('enter text here')
 
 if text:
+    messages = [
+        {"role": "user", "content": text},
+    ]
     out = pipe(text)
     st.json(out)