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
 ============
@@ -120,7 +120,7 @@ def predict(content, policy):
 
 
 # Create the interface
-with gr.Blocks(api_name=False) as demo:
+with gr.Blocks() as demo:
     with gr.Row():
         # Left column with inputs
         with gr.Column(scale=1):
@@ -167,7 +167,8 @@ with gr.Blocks(api_name=False) as demo:
     submit_btn.click(
         fn=predict,
         inputs=[input1, input2],
-        outputs=output
+        outputs=output,
+        api_name=False
     )
 
 # Launch the interface