merge

app.py

@@ -1,7 +1,5 @@
-from __future__ import annotations
 import spaces
 import gradio as gr
-import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import whoami
 import os
@@ -80,23 +78,5 @@ demo = gr.ChatInterface(
     # chatbot_style="default"
 )
 
-def hello(profile: gr.OAuthProfile | None) -> str:
-    if profile is None:
-        return "I don't know you."
-    return f"Hello {profile.name}"
-
-def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
-    if oauth_token is None:
-        return "Please deploy this on Spaces and log in to list organizations."
-    org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
-    return f"You belong to {', '.join(org_names)}."
-
-with gr.Blocks() as demo:
-    gr.LoginButton()
-    m1 = gr.Markdown()
-    m2 = gr.Markdown()
-    demo.load(hello, inputs=None, outputs=m1)
-    demo.load(list_organizations, inputs=None, outputs=m2)
-
 if __name__ == "__main__":
     demo.launch()