Update app.py

app.py

@@ -3,12 +3,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import spaces
 
 # Load model and tokenizer
-
+model_name = "yuchenlin/Rex-v0.1-1.5B"
 
 device = "cuda" # the device to load the model onto
-tokenizer = AutoTokenizer.from_pretrained("yuchenlin/Rex-v0.1-1.5B", trust_remote_code=True, rex_size=3)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, rex_size=3)
 model = AutoModelForCausalLM.from_pretrained(
-    "yuchenlin/Rex-v0.1-1.5B",
+    model_name,
     torch_dtype="auto"
 )
 model.to(device)
@@ -73,4 +73,4 @@ demo = gr.ChatInterface(
 
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch(share=True)