Update app.py

app.py

@@ -7,7 +7,7 @@ import torch
 import torch.nn.functional as F
 from peft import PeftConfig, PeftModel
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
-"""
+
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 base_model_name = "google/gemma-2-9b"
@@ -27,7 +27,7 @@ model.merge_and_unload()
 model = model.to(device)
 
 tokenizer = AutoTokenizer.from_pretrained(base_model_name)
-"""
+
 PROMPT = """
 INSTRUCTIONS
 ============
@@ -126,7 +126,6 @@ iface = gr.Interface(
             gr.Textbox(label="Policy", lines=8, value=DEFAULT_POLICY)], 
     outputs=[gr.Textbox(label="Result"),
              gr.Markdown("A BUNCH OF STUFF")],
-    article="TEST CONTENT",
     title="Zentropi CoPE Demo",
     #description="See if the given content violates your given policy.",
     api_name=False