update model

app.py

@@ -19,12 +19,8 @@ import os
 from langchain.schema import HumanMessage, SystemMessage
 from langchain_openai import ChatOpenAI
 
-
-# In[18]:
-
-
 # Set the model's file path
-file_path = Path("models/model_adam_scaled.h5")
+file_path = Path("models/model_adam_5.h5")
 
 # Load the model to a new object
 adam_5 = tf.keras.models.load_model(file_path)
@@ -37,7 +33,6 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
 
 print(f"OpenAI API Key Loaded: {openai_api_key is not None}")
 
-
 # Load the model and tokenizer for translation
 model = MBartForConditionalGeneration.from_pretrained(
     "facebook/mbart-large-50-many-to-many-mmt"
@@ -49,10 +44,6 @@ tokenizer = MBart50TokenizerFast.from_pretrained(
 # Set source language
 tokenizer.src_lang = "en_XX"
 
-
-# In[22]:
-
-
 # Constants
 # Language information MBart
 language_info = [
@@ -129,19 +120,12 @@ default_language = "English"
 malignant_text = "Malignant. Please consult a doctor for further evaluation."
 benign_text = "Benign. Please consult a doctor for further evaluation."
 
-
-# In[23]:
-
-
 # Create instance
 llm = ChatOpenAI(
     openai_api_key=openai_api_key, model_name="gpt-3.5-turbo", temperature=0
 )
 
 
-# In[24]:
-
-
 # Method to get system and human messages for ChatOpenAI - Predictions
 def get_prediction_messages(prediction_text):
     # Create a HumanMessage object
@@ -156,9 +140,6 @@ def get_prediction_messages(prediction_text):
     return [system_message, human_message]
 
 
-# In[25]:
-
-
 # Method to get system and human messages for ChatOpenAI - Help
 def get_chat_messages(chat_prompt):
     # Create a HumanMessage object
@@ -172,9 +153,6 @@ def get_chat_messages(chat_prompt):
     return [system_message, human_message]
 
 
-# In[26]:
-
-
 # Method to predict the image
 def predict_image(language, img):
     try:
@@ -264,9 +242,6 @@ def predict_image(language, img):
         )
 
 
-# In[27]:
-
-
 # Method for on submit
 def on_submit(language, img):
     print(f"Language: {language}")
@@ -283,9 +258,6 @@ def on_submit(language, img):
     return predict_image(language, img)
 
 
-# In[28]:
-
-
 # Method for on clear
 def on_clear():
     return (
@@ -299,9 +271,6 @@ def on_clear():
     )
 
 
-# In[29]:
-
-
 # Method for on chat
 def on_chat(language, chat_prompt):
     try:
@@ -351,9 +320,6 @@ def on_chat(language, chat_prompt):
         )
 
 
-# In[30]:
-
-
 # Gradio app
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="green")) as demo: