Update app.py

app.py

@@ -4,12 +4,27 @@ from sentence_transformers import SentenceTransformer, util
 import os
 import requests
 
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForCodeGeneration, AutoTokenizerForCodeGeneration
 
-model_name = "acecalisto3/InstructiPhi"  # You can change this to other models if desired
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+# Load the pre-trained model and tokenizer
+model_name = "OpenBMB/multilingual-codeparrot"
+model = AutoModelForCodeGeneration.from_pretrained(model_name)
+tokenizer = AutoTokenizerForCodeGeneration.from_pretrained(model_name)
 
+# Define input prompt
+input_prompt = "(input value = highest-level-quality code content invocation ; True)"
+
+# Tokenize the input prompt
+input_ids = tokenizer(input_prompt, return_tensors="pt", truncation=True)
+
+# Generate the code
+generated_code = model.generate(input_ids.to(model.device))
+
+# Decode the generated code
+generated_code_str = tokenizer.batch_decode(generated_code, skip_special_tokens=True)[0]
+
+# Print the generated code
+print(generated_code_str)
 # Constants for enhanced organization
 GITHUB_API_BASE_URL = "https://api.github.com/repos"
 DEFAULT_MODEL = "apple/OpenELM"
@@ -161,7 +176,7 @@ with gr.Blocks() as demo:
         choices=[
             "‎apple/OpenELM",
             "Gabriel/Swe-review-setfit-model",
-            "acecalisto3/InstructiPhi"
+            "OpenBMB/multilingual-codeparrot"
         ],
         label="Select Model for Issue Resolution",
         value=DEFAULT_MODEL,