Update app.py

app.py

@@ -62,7 +62,6 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
     input_tokens = len(tokenizer.encode(prompt))
     total_tokens_used += input_tokens
     available_tokens = 32768 - total_tokens_used
-
     if available_tokens <= 0:
         yield f"Error: 입력이 최대 허용 토큰 수를 초과합니다. Total tokens used: {total_tokens_used}"
         return
@@ -70,8 +69,16 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
     formatted_prompt = format_prompt(prompt, history)
     output_accumulated = ""
     try:
-        stream = client.text_generation(formatted_prompt, temperature=temperature, max_new_tokens=min(max_new_tokens, available_tokens),
-                                        top_p=top_p, repetition_penalty=repetition_penalty, do_sample=True, seed=42, stream=True)
+        stream = client.text_generation(
+            formatted_prompt,
+            temperature=temperature,
+            max_new_tokens=min(max_new_tokens, available_tokens),
+            top_p=top_p,
+            repetition_penalty=repetition_penalty,
+            do_sample=True,
+            seed=42,
+            stream=True
+        )
         for response in stream:
             output_part = response['generated_text'] if 'generated_text' in response else str(response)
             output_accumulated += output_part
@@ -96,7 +103,6 @@ def process_financial_data(ticker):
     valid, message = validate_ticker(ticker)
     if not valid:
         return f"Error: {message} - 주식 티커 '{ticker}'를 확인하세요."
-
     try:
         stock = yf.Ticker(ticker)
         real_time_price = stock.history(period="1d")
@@ -119,20 +125,20 @@ mychatbot = gr.Chatbot(
     likeable=True,
 )
 
-
 examples = [
-    ["반드시 한글로 답변할것.", []],  # history 값을 빈 리스트로 제공    
+    ["반드시 한글로 답변할것.", []],  # history 값을 빈 리스트로 제공
     ["분석 결과 보고서 다시 출력할것", []],
     ["추천 종목 알려줘", []],
     ["그 종목 투자 전망 예측해", []]
 ]
 
-
 css = """
 h1 {
     font-size: 14px; /* 제목 글꼴 크기를 작게 설정 */
 }
-footer {visibility: hidden;}
+footer {
+    visibility: hidden;
+}
 """
 
 demo = gr.ChatInterface(
@@ -142,7 +148,7 @@ demo = gr.ChatInterface(
     retry_btn=None,
     undo_btn=None,
     css=css,
-    examples=examples    
+    examples=examples
 )
 
 demo.queue().launch(show_api=False)