Update app.py

app.py

@@ -1,5 +1,5 @@
-from transformers import AutoModel, AutoTokenizers
-import gradio as gr
+import gradio as gr 
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 sys_message = """
 This model can generate untruths, lies or inappropriate things. Only for testing and validation. 
@@ -16,9 +16,8 @@ generation_config = {
     "stop_token": '### Instruction:'
 }
 
-llm = AutoModel.from_pretrained("tevykuch/sftsl0th")
-# llm = AutoModelForCausalLM.from_pretrained("tevykuch/sl0th", hf=True)
-tokenizer = AutoTokenizer.from_pretrained(llm)
+tokenizer = AutoTokenizer.from_pretrained("tevykuch/sftsl0th")
+llm = AutoModelForCausalLM.from_pretrained("tevykuch/sftsl0th")
 
 def stream(prompt):
     # Tokenize the prompt