Update app.py

app.py

@@ -1,23 +1,20 @@
 import streamlit as st
 from transformers import pipeline
+pipe = pipeline('text-generation', model='Pclanglais/MonadGPT', device='cuda')
 
-# Load the text classification pipeline for sentiment analysis
-fake_news_detector = pipeline("text-classification", model="distilroberta-base-finetuned-fake-news-detection")
+def main():
+  st.title("MonadGPT Streamlit App")
+  st.markdown("This is a streamlit app that allows you to have a conversation abbout any topic")
 
-# Streamlit app
-st.title("Fake News Detector")
+  user_input = st.text_input("Your question:")
 
-# Input text box for user-provided news article
-news_article = st.text_area("Enter the news article:", "")
+  if user_input:
+    response = pipe(user_input, max_length=256, do_sample=True, top_k=50, top_p=0.95, early_stopping=True)
+    st.write(response[0]['generated_text'])
 
-# Check fake news button
-if st.button("Check Fake News"):
-    if news_article:
-        # Perform fake news detection
-        result = fake_news_detector(news_article)[0]
+  if st.button("Restart"):
+    st.empty()
+
+if __name__ == "__main__":
+  main()
 
-        # Display result
-        st.subheader("Result:")
-        st.write(f"Label: {result['label']}, Confidence: {result['score']:.2f}")
-    else:
-        st.warning("Please enter a news article for analysis.")