Update main.py

main.py

@@ -7,8 +7,16 @@ import gradio as gr
 tokenizer = AutoTokenizer.from_pretrained("huggyllama/llama-7b")
 model = AutoModelForCausalLM.from_pretrained("huggyllama/llama-7b")
 
+print("***")
+print("Loaded tokenizer and model")
+print("***")
+
 pipe_flan = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
+print("***")
+print("Created pipeline")
+print("***")
+
 # Text generation
 def generator(input):
     output = pipe_flan(input, max_length=50, num_return_sequences=1)
@@ -21,5 +29,10 @@ demo = gr.Interface(
     outputs=gr.outputs.Textbox(label="Generated Text")
 )
 
+host, port = "0.0.0.0", 7860
+print("***")
+print(f"Set up interface. Hosting now on {host}:{port}")
+print("***")
+
 # Lauching the Gradio Interface
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name=host, server_port=port)