Update app.py

app.py

@@ -3,12 +3,9 @@ import os
 os.system("pip install transformers")
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-generated_text = tokenizer.decode(output[0])
-print(generated_text)
-
 def GenerateResp(prompt):
     model = AutoModelForCausalLM.from_pretrained('rexwang8/qilin-lit-6b')
-    tokenizer = AutoTokenizer.from_pretrained('rexwang8/lit-6b')
+    tokenizer = AutoTokenizer.from_pretrained('rexwang8/qilin-lit-6b')
     
     #prompt = '''I had eyes but couldn't see Mount Tai!'''