Update app.py

app.py

@@ -57,27 +57,24 @@ def generate_text(prompt="Who is the CEO of Apple?"):
     # cleaned_output_text = output_text.replace(prompt, "")
     # return cleaned_output_text
 
-def main():
-    os.getenv("hf_token")
-    description = "Zephyr-beta"
 
-    examples = [
-        ["What is the capital of France?", "The capital of France is Paris."],
-        [
-            "Who wrote the novel 'Pride and Prejudice'?",
-            "The novel 'Pride and Prejudice' was written by Jane Austen.",
-        ],
-        ["What is the square root of 64?", "The square root of 64 is 8."],
-    ]
+os.getenv("hf_token")
+description = "Zephyr-beta"
 
-    gradio_interface = gr.Interface(
-        fn=generate_text,
-        inputs="text",
-        outputs="text",
-        examples=examples,
-        title="Zephyr-B",
-    )
-    gradio_interface.launch()
+examples = [
+    ["What is the capital of France?", "The capital of France is Paris."],
+    [
+        "Who wrote the novel 'Pride and Prejudice'?",
+        "The novel 'Pride and Prejudice' was written by Jane Austen.",
+    ],
+    ["What is the square root of 64?", "The square root of 64 is 8."],
+]
 
-if __name__ == "__main__":
-    main()
+gradio_interface = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    examples=examples,
+    title="Zephyr-B",
+)
+gradio_interface.launch()