Update app.py

app.py

@@ -8,19 +8,18 @@ load_dotenv()
 
 OLLAMA_BASE_URL = os.getenv("https://alpaca-upright-vulture.ngrok-free.app")
 
-llm = ChatOllama(model="llama3.2", base_url = OLLAMA_BASE_URL, temperature=0.5)
+llm = ChatOllama(model="llama3", base_url = OLLAMA_BASE_URL, temperature=0.5)
 
-#prompt = ChatPromptTemplate.from_template("Tell me a joke about a {subject}")
 prompt = ChatPromptTemplate.from_messages(
     [
-        ("system","Generate a list of synonyms for the following word, only give 5 words and no extra messages."),
+        ("system","Generate a list of synonyms for the following word, only give 5 words with examples."),
         ("human","{input}")
     ]
 )
 
 def synonyms(text):
-    response = chain.invoke({"input":text})
-    return response.content
+    response = chain.invoke({"input": text})
+    return response.content 
 
 chain = prompt | llm
 
@@ -29,7 +28,7 @@ demo = gr.Interface(
     inputs = ["text"],
     outputs = ["text"],
     title = "Synonym Generator",
-    description = "Generate a list of synonyms for the following word, only give 5 words with examples i.e use word in the sentences."
+    description = "Generate a list of synonyms for the following word"
 )
 
 demo.launch()