Update app.py

app.py

@@ -10,9 +10,11 @@ OAI_API_KEY=os.getenv("OPENAI_API_KEY")
 #login(token=HUGGINGFACEHUB_API_TOKEN)
 
 # tokenizer for generating prompt
+print ("Tokenizer")
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-70b-chat-hf")
 
 # inference client
+print ("Inf.Client")
 client = InferenceClient("https://api-inference.huggingface.co/models/meta-llama/Llama-2-70b-chat-hf")
 
 # generate function
@@ -26,6 +28,7 @@ def generate(text):
                     top_p=0.9,
                     temperature=0.6,
                 )
+    print (res)
     return res.strip()
 
 # test client