Update app.py

app.py

@@ -24,7 +24,7 @@ if not torch.cuda.is_available():
 
 # If a GPU is available, load the model and tokenizer with specific configurations.
 if torch.cuda.is_available():
-    model_id = "mahiatlinux/MasherAI-7B-v6.1-another_test3"
+    model_id = "mahiatlinux/MasherAI-v6.1-7B-checkpoint3-code4"
     model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", load_in_4bit=True)
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
@@ -44,7 +44,7 @@ def generate(
     # Preparing conversation history for processing.
     conversation = []
     # Adding system prompt.
-    #conversation.append({"from": "human", "value": system_prompt})
+    # conversation.append({"from": "human", "value": system_prompt})
     # Extending the conversation history with user and assistant interactions.
     for user, assistant in chat_history:
         conversation.extend([{"from": "human", "value": user}, {"from": "gpt", "value": assistant}])