Update app.py

app.py

@@ -21,27 +21,26 @@ pipe = pipeline(
     do_sample=False,
     num_beams=4,
     repetition_penalty=1.5,
-    return_full_text=False
+    return_full_text=False,
+    eos_token_id=tokenizer.eos_token_id
 )
 
 # Prompt template
 prompt_template = """
 You are a financial market analyst.
 Before making a prediction, you always analyze the past, which is provided in the Context below.
-
-When analyzing the Context, pay close attention to the dates and the economic conditions of that period (e.g., interest rate changes, inflation trends, sector performance, or geopolitical events).
+When analyzing the Context, pay close attention to the dates and the economic conditions of that period 
+(e.g., interest rate changes, inflation trends, sector performance, or geopolitical events).
 Base your reasoning on what actually happened during that specific timeframe.
 
-If, in the past, multiple stocks have changed, provide a list of those stocks along with their price changes.
-If a specific event led to different stock price reactions under varying circumstances (for example, depending on whether interest rates were rising or falling), describe each plausible scenario and explain the reasoning behind the reaction in each case.
+If, in the past, given an event, multiple stocks have changed, provide a list of those stocks along with their price changes.
+If a specific event led to different stock price reactions under varying circumstances 
+(for example, depending on whether interest rates were rising or falling), describe each plausible scenario and explain the reasoning behind the reaction in each case.
 
 Please respond with:
-
-Chosen Stock or Stocks: (name/names)
-
-Prediction(s): (expected price change)
-
-Explanation: (brief, clear, and based on past patterns — mention the time period and the factors influencing the outcome)
+-Chosen Stock or list of Stocks: (name/names)
+-Prediction(s): (expected price change)
+-Explanation: (brief, clear, and based on past patterns — mention the time period and the factors influencing the outcome)
 
 Context:
 {context}
@@ -82,7 +81,8 @@ examples = [
     "Trump imposes tariffs",
     "Tesla announces a new affordable electric vehicle model",
     "Nvidia releases new GPU technology",
-    "Apple launches Apple TV+ subscription service"
+    "Apple launches Apple TV+ subscription service",
+    "Elon Musk created a new political party against Trump"
 ]
 
 description_md = """