fixes

gemmademo/_model.py

@@ -44,15 +44,15 @@ class LlamaCppGemmaModel:
             "description": "2B parameters, instruction-tuned",
             "type": "instruct",
         },
-        "gemma-7b-it": {
-            "model_path": "models/gemma-7b-it.gguf",
+        "gemma-7b": {
+            "model_path": "models/gemma-7b.gguf",
             "repo_id": "MaziyarPanahi/gemma-7b-GGUF",  # update to the actual repo id
             "filename": "gemma-7b.Q4_K_M.gguf",  # update to the actual filename
             "description": "7B parameters, instruction-tuned",
             "type": "instruct",
         },
-        "gemma-7b": {
-            "model_path": "models/gemma-7b.gguf",
+        "gemma-7b-it": {
+            "model_path": "models/gemma-7b-it.gguf",
             "repo_id": "rahuldshetty/gemma-7b-it-gguf-quantized",  # repository for the GGUF model
             "filename": "gemma-7b-it-Q4_K_M.gguf",  # updated filename for GGUF model
             "description": "7B parameters in GGUF format",

gemmademo/_prompts.py

@@ -36,8 +36,6 @@ class PromptManager:
         """
         Format user input for code completion task
         """
-        prompt = f"""Complete the following code snippet with proper syntax and best practices:
-        {user_input}
-
-        Completed code:"""
+        prompt = f"""Complete the following code snippet directly without explanations or extra text:
+        {user_input}"""
         return prompt