Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
+# Load model and tokenizer from local files in the root (already uploaded)
 tokenizer = AutoTokenizer.from_pretrained(".", local_files_only=True)
 model = AutoModelForSeq2SeqLM.from_pretrained(".", local_files_only=True)
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
-def predict(task, prompt, context="", auto_cot=False):
+# Inference function
+def predict(task, prompt, context, auto_cot):
     full_prompt = f"[TASK: {task.upper()}] {prompt}"
     if context:
         full_prompt += f" Context: {context}"
@@ -14,10 +16,12 @@ def predict(task, prompt, context="", auto_cot=False):
     output = pipe(full_prompt, max_new_tokens=128)[0]["generated_text"]
     return output
 
+# Create Interface
 demo = gr.Interface(
     fn=predict,
     inputs=["text", "text", "text", "checkbox"],
     outputs="text"
 )
 
-demo.launch()
+# ⚠️ Do NOT set share=True or inbrowser=True
+demo.launch()