Update app.py

app.py

@@ -2,7 +2,7 @@
 
 import gradio as gr
 import os
-"""
+
 import torch
 import torch.nn.functional as F
 from peft import PeftConfig, PeftModel
@@ -27,7 +27,7 @@ model.merge_and_unload()
 model = model.to(device)
 
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
-"""
+
 PROMPT = """
 INSTRUCTIONS
 ============
@@ -94,7 +94,6 @@ DEFAULT_CONTENT = "Put your content sample here."
 
 # Function to make predictions
 def predict(content, policy):
-    return "TEST"
     input_text = PROMPT.format(policy=policy, content=content)
     input_ids = tokenizer.encode(input_text, return_tensors="pt")
 
@@ -122,7 +121,7 @@ def predict(content, policy):
 # Create the interface
 
 with gr.Blocks() as demo:
-    gr.Markdown("# My Interactive Interface")
+    gr.Markdown("# Zentropi CoPE Demo")
 
     with gr.Row():
         # Left column with inputs
@@ -135,21 +134,21 @@ with gr.Blocks() as demo:
             output = gr.Textbox(label="Result")
             submit_btn = gr.Button("Submit")
             notes = gr.Markdown("""
-    # Usage Instructions
-    
-    This interface allows you to process two inputs and see the results.
-    
-    ## Features
-    
-    1. Two input text boxes for versatile data entry
-    2. Real-time processing
-    3. Clear output display
-    
-    ## How to Use
-    
-    1. Enter your first input in the left text box
-    2. Enter your second input in the right text box
-    3. Click the "Process" button to see results
+## About CoPE
+
+CoPE (the COntent Policy Evaluation engine) is a small language model capable of accurate content policy labeling. This is a **demo* of our initial release and should **NOT** be used for any production use cases.
+
+## How to Use
+                                                                                                                                                                                                                                                                   
+1. Enter your content in the "Content" box.
+2. Specify your policy in the "Policy" box.
+3. Click "Submit" to see the results.
+
+## More Info
+
+- [Give us feedback](https://forms.gle/BHpt6BpH2utaf4ez9) to help us improve
+- [Read our FAQ](https://docs.google.com/document/d/1Cp3GJ5k2I-xWZ4GK9WI7Xv8TpKdHmjJ3E9RbzP5Cc_Y/edit) to learn more about CoPE
+- [Join our mailing list](https://forms.gle/PCABrZdhTuXE9w9ZA) to keep in touch
             """)
     
     # Button below inputs