Update app.py

app.py

@@ -2,11 +2,12 @@ import os
 from threading import Thread
 from typing import Iterator
 import os
-from huggingface_hub import login
+from huggingface_hub import login,whoami
 import gradio as gr
 import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+import argparse
 
 MAX_MAX_NEW_TOKENS = 128
 DEFAULT_MAX_NEW_TOKENS = 1024
@@ -15,7 +16,11 @@ model = None
 tokenizer = None
 
 my_token = os.getenv("HF_AUTH_TOKEN")
-login(token = my_token)
+
+try:
+    username = whoami()
+except OSError:
+    login(token = my_token, add_to_git_credential = True)
 
 model_id = "stabilityai/ar-stablelm-2-chat"
 model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", trust_remote_code=True)
@@ -135,4 +140,7 @@ with gr.Blocks(css_paths="style.css", fill_height=True) as demo:
     
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    parser = argparse.ArgumentParser(description="Gradio App with Sharing")
+    parser.add_argument("--share", action="store_true", help="Enable public sharing") 
+    args = parser.parse_args()
+    demo.queue(max_size=20).launch(share = args.share)