Update app.py

app.py

@@ -8,9 +8,7 @@ MODEL_PATH = "ddfws/Rezaeian-StatsAI"
 
 print("Loading tokenizer...")
 
-tokenizer = AutoTokenizer.from_pretrained(
-    MODEL_PATH
-)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 
 
 print("Loading model...")
@@ -18,7 +16,7 @@ print("Loading model...")
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_PATH,
     device_map="auto",
-    dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )
 
 model.eval()
@@ -26,20 +24,20 @@ model.eval()
 print("MODEL READY")
 
 
+
 def format_text(text):
 
-    # فرمول های رایج آماری
     replacements = {
         "β0": r"$\beta_0$",
         "β1": r"$\beta_1$",
         "σ2": r"$\sigma^2$",
         "R2": r"$R^2$",
-        "xbar": r"$\bar{x}$",
-        "mu": r"$\mu$",
+        "x̄": r"$\bar{x}$",
+        "mu": r"$\mu$"
     }
 
     for a,b in replacements.items():
-        text = text.replace(a,b)
+        text=text.replace(a,b)
 
     return text
 
@@ -51,27 +49,21 @@ def generate(message, history):
 
     for user, bot in history:
 
-        messages.append(
-            {
-                "role":"user",
-                "content":user
-            }
-        )
+        messages.append({
+            "role":"user",
+            "content":user
+        })
 
-        messages.append(
-            {
-                "role":"assistant",
-                "content":bot
-            }
-        )
+        messages.append({
+            "role":"assistant",
+            "content":bot
+        })
 
 
-    messages.append(
-        {
-            "role":"user",
-            "content":message
-        }
-    )
+    messages.append({
+        "role":"user",
+        "content":message
+    })
 
 
     prompt = tokenizer.apply_chat_template(
@@ -87,157 +79,98 @@ def generate(message, history):
     ).to(model.device)
 
 
-
     with torch.no_grad():
 
-        output = model.generate(
+        output=model.generate(
             **inputs,
             max_new_tokens=700,
             temperature=0.6,
             top_p=0.9,
-            do_sample=True,
-            repetition_penalty=1.1
+            do_sample=True
         )
 
 
-
-    answer = tokenizer.decode(
+    answer=tokenizer.decode(
         output[0][inputs.input_ids.shape[1]:],
         skip_special_tokens=True
     )
 
 
-    answer = format_text(answer)
-
-
-    return answer
-
-
+    return format_text(answer)
 
 
-css = """
 
-/* حذف موارد اضافی */
+css="""
 
-footer {
-    display:none !important;
+footer{
+display:none !important;
 }
 
 
-.gradio-container {
-
-    max-width:1100px !important;
-    direction:rtl;
-
+.gradio-container{
+direction:rtl;
 }
 
 
-/* پیام ها */
-
-.message {
-
-    direction:rtl !important;
-    text-align:right !important;
-    unicode-bidi:plaintext;
-
+.message{
+direction:rtl !important;
+text-align:right !important;
+unicode-bidi:plaintext;
 }
 
 
-/* متن فارسی */
-
-.markdown {
-
-    direction:rtl;
-    text-align:right;
-
+.markdown{
+direction:rtl;
+text-align:right;
 }
 
 
-/* ورودی */
-
-textarea {
-
-    direction:rtl !important;
-    text-align:right !important;
-
+textarea{
+direction:rtl !important;
+text-align:right !important;
 }
 
 
-/* فرمول ها */
-
-.katex {
-
-    direction:ltr !important;
-
+code,pre{
+direction:ltr !important;
+text-align:left !important;
 }
 
 
-.MathJax {
-
-    direction:ltr !important;
-
-}
-
-
-/* کد و انگلیسی */
-
-code, pre {
-
-    direction:ltr !important;
-    text-align:left !important;
-
-}
-
 """
 
 
-head = """
+head="""
 
 <script>
-window.MathJax = {
- tex: {
-   inlineMath: [['$', '$']],
-   displayMath: [['$$','$$']]
- }
+window.MathJax={
+tex:{
+inlineMath:[['$','$']],
+displayMath:[['$$','$$']]
+}
 };
 </script>
 
-<script 
-src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js">
-</script>
+<script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
 
 """
 
 
-demo = gr.ChatInterface(
+chatbot = gr.Chatbot(
+    height=600
+)
 
-    fn=generate,
 
+demo = gr.ChatInterface(
+    fn=generate,
+    chatbot=chatbot,
     title="Rezaeian StatsAI",
-
-    description=
-    "دستیار هوش مصنوعی آمار و احتمال مهندسی",
-
+    description="دستیار هوش مصنوعی آمار و احنمال مهندسی",
     css=css,
-
-    head=head,
-
-    chatbot=gr.Chatbot(
-        height=650,
-        show_copy_button=True
-    ),
-
-    examples=[
-        "فرمول میانگین نمونه را بنویس",
-        "رگرسیون خطی را با فرمول توضیح بده",
-        "آزمون t چیست؟",
-        "انحراف معیار را محاسبه کن"
-    ]
-
+    head=head
 )
 
 
-
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,