Update app.py

app.py

@@ -1,34 +1,10 @@
-import streamlit as st
-from transformers import pipeline
+import gradio as gr
 
-# Load the sentiment analysis pipeline
-sentiment_classifier = pipeline("text-classification", model="isom5240grp21/finetuned_model1")
 
-# Load the keyword extraction pipeline
-keyword_extractor = pipeline("text2text-generation", model="ilsilfverskiold/bart_keywords")
+def greet(name):
+    return "Hello " + name
 
-def main():
-    st.title("Hotel Review Keywords Extractor")
 
-    # User input for the review
-    review = st.text_area("Enter your review:")
-
-    if st.button("Analyze"):
-        if review:
-            # Perform sentiment analysis
-            sentiment = sentiment_classifier(review)
-            sentiment_label = sentiment[0]['label']
-
-            # Perform keyword extraction
-            keywords = keyword_extractor(review)
-            generated_text = keywords[0]['generated_text']
-
-            # Display the result based on sentiment
-            if sentiment_label == 'NEGATIVE':  
-                st.write(f"The hotel did bad in: {generated_text}")
-            else: 
-                st.write(f"The hotel did good in: {generated_text}")
-
-if __name__ == "__main__":
-    main()
+demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
+demo.launch()