Update app.py

app.py

@@ -1,10 +1,11 @@
 import random
 import gradio as gr
+from transformers import AutoModelForCausalLM
 
-def alternatingly_agree(message, history):
-    if len(history) % 2 == 0:
-        return f"Yes, I do think that '{message}'"
-    else:
-        return "I don't think so"
+def LMmodel(message, history):
+    base_model_id="LeoLM/leo-mistral-hessianai-7b-chat"
+    base_model = "AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto",trust_remote_code=True)
+    return works
+    
 
-gr.ChatInterface(alternatingly_agree).launch()
+gr.ChatInterface(LMmodel).launch()