Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from IPython.display import IFrame
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
 model_name = "NimaKL/FireWatch_tiny_75k"
@@ -12,6 +13,17 @@ def predict(text):
     label_id = logits.argmax(axis=1).item()
     return "Danger of fire hazard!" if label_id == 1 else "It is unlikely that a fire will start in this area."
 
+# Define a custom CSS style
+custom_style = """
+    body {
+        background-color: #F262626;
+    }
+"""
+
+# Define the function to display the Google Sheets document
+def show_sheet():
+    return IFrame("https://docs.google.com/spreadsheets/d/1b2aAZ8ue5NI7hZ-MiNc8y4dFthbrRiXlsaEeYSHgNCM/edit?usp=sharing", width=800, height=600)
+
 io = gr.Interface(
     fn=predict,
     inputs="text",
@@ -19,10 +31,14 @@ io = gr.Interface(
     title="FireWatch",
     description="Predict whether a data row describes a fire hazard or not.",
     output_description="Prediction",
-    examples=[['-26.76123, 147.15512, 393.02, 203.63'], ['-26.7598, 147.14514, 361.54, 79.4'], ['-25.70059, 149.48932, 313.9, 5.15'], ['-24.4318, 151.83102, 307.98, 8.79'], ['-23.21878, 148.91298, 314.08, 7.4'], ['7.87518, 19.9241, 316.32, 39.63'], ['-20.10942, 148.14326, 314.39, 8.8'], ['7.87772, 19.9048, 304.14, 13.43'], ['-20.79866, 124.46834, 366.74, 89.06']]
-,
+    examples=[['-26.76123, 147.15512, 393.02, 203.63'], ['-26.7598, 147.14514, 361.54, 79.4'], ['-25.70059, 149.48932, 313.9, 5.15'], ['-24.4318, 151.83102, 307.98, 8.79'], ['-23.21878, 148.91298, 314.08, 7.4'], ['7.87518, 19.9241, 316.32, 39.63'], ['-20.10942, 148.14326, 314.39, 8.8'], ['7.87772, 19.9048, 304.14, 13.43'], ['-20.79866, 124.46834, 366.74, 89.06']],
     output_component_names=["Prediction"],
-    theme="Streamlit"
+    theme="Streamlit",
+    css=custom_style,
+    article="<h2>FireWatch App</h2><p>This app predicts whether a data row describes a fire hazard or not. The prediction is based on a machine learning model that has been trained on a dataset of data rows.</p><h2>Data Rows Sheet</h2>"
 )
 
+# Add the IFrame component to the app
+io.add_component("Sample Data Sheet", gr.outputs.HTML, show_sheet)
+
 io.launch()