refactor

app.py

@@ -5,16 +5,16 @@ import torch
 import transformers
 from transformers import AutoTokenizer
 
-logging.info("APP startup")
+logging.basicConfig(level=logging.INFO)
+
 print("APP startup")
 
 pipe_flan = transformers.pipeline("text2text-generation", model="google/flan-t5-small")
 def google_flan(input_text, request: gr.Request):
     print("New response 2")
     print(request.query_params)
-    print(request.query_params["name"])
-    print(os.environ["HF_TOKEN"][:5])
-    logging.info(os.environ["HF_TOKEN"][:5])
+    print(os.environ.get("HF_TOKEN")[:5])
+    logging.info(os.environ.get("HF_TOKEN")[:5])
     return pipe_flan(input_text)
 
 # model = "meta-llama/Llama-2-7b-chat-hf"