Update app.py

app.py

@@ -1,13 +1,24 @@
 import gradio as gr
 
-# Load the model
-model_interface = gr.load("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+# Load the model from Hugging Face
+def load_model():
+    try:
+        model_interface = gr.load("huggingface/TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+        return model_interface
+    except ValueError as e:
+        print(f"Error loading the model: {e}")
+        return None
 
 # Create a Gradio interface with custom title and logo
 def main():
+    model_interface = load_model()
+    if model_interface is None:
+        print("Failed to load the model. Exiting.")
+        return
+
     with gr.Blocks() as demo:
         # Add the title
-        gr.Markdown("# Tiny llama 1B Demo")
+        gr.Markdown("# TinyLlama Demo")
         
         # Add the logo
         gr.Image("orYx logo.png", elem_id="logo", show_label=False, interactive=False)
@@ -19,4 +30,4 @@ def main():
     demo.launch()
 
 if __name__ == "__main__":
-    main()
+    main()