Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from typing import List, Tuple
-import os
 from gradio_client import Client
 
 def create_chat_app():
@@ -13,11 +12,11 @@ def create_chat_app():
         top_p: float,
     ) -> str:
         """
-        Process user message and generate a response using the Llama model.
+        Process user message and generate a response using the Llama 3.3 70B model.
         """
         try:
-            # Initialize client for the specific space
-            client = Client("1ofteamos/meta-llama-Llama-3.2-1B-Instruct")
+            # Initialize client for the newer API
+            client = Client("aifeifei798/feifei-chat")
             
             # Format the conversation history and current message
             formatted_message = f"{system_message}\n\nConversation history:\n"
@@ -29,9 +28,18 @@ def create_chat_app():
             
             formatted_message += f"User: {message}"
             
-            # Get response from the model
+            # Prepare the message payload
+            message_payload = {
+                "text": formatted_message,
+                "files": []
+            }
+            
+            # Get response from the model with the specified parameters
             response = client.predict(
-                message=formatted_message,
+                message=message_payload,
+                feifei_select=True,
+                additional_dropdown="meta-llama/Llama-3.3-70B-Instruct",
+                image_mod="pixtral",
                 api_name="/chat"
             )
             
@@ -43,9 +51,9 @@ def create_chat_app():
     # Interface configuration
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("""
-        # 🤖 Chat com Llama em Português
+        # 🤖 Chat com Llama 3.3 70B em Português
         
-        Este é um chatbot baseado no modelo Llama. Para usar:
+        Este é um chatbot baseado no modelo Llama 3.3 70B. Para usar:
         1. Digite sua mensagem no campo abaixo
         2. Ajuste os parâmetros conforme necessário
         3. Pressione Enter para enviar
@@ -55,19 +63,19 @@ def create_chat_app():
             respond,
             additional_inputs=[
                 gr.Textbox(
-                    value="Você é um assistente amigável e prestativo que responde em português.",
+                    value="Você é um assistente amigável e prestativo que responde em português. Você é baseado no modelo Llama 3.3 70B.",
                     label="Mensagem do Sistema"
                 ),
                 gr.Slider(
                     minimum=1,
-                    maximum=2048,
-                    value=512,
+                    maximum=4096,
+                    value=2048,
                     step=1,
                     label="Máximo de Tokens"
                 ),
                 gr.Slider(
                     minimum=0.1,
-                    maximum=4.0,
+                    maximum=2.0,
                     value=0.7,
                     step=0.1,
                     label="Temperatura"
@@ -80,22 +88,24 @@ def create_chat_app():
                     label="Top-p (Amostragem Nucleus)"
                 ),
             ],
-            title="Chat com Llama",
-            description="Um chatbot interativo usando o modelo Llama.",
+            title="Chat com Llama 3.3 70B",
+            description="Um chatbot interativo usando o modelo Llama 3.3 70B Instruct.",
             examples=[
                 ["Olá! Como você está?"],
                 ["Pode me explicar o que é inteligência artificial?"],
-                ["Qual é a capital do Brasil?"]
+                ["Qual é a capital do Brasil?"],
+                ["Me ajude a escrever um código em Python para calcular fibonacci."]
             ]
         )
         
         gr.Markdown("""
         ### ℹ️ Informações
-        - Modelo: Llama 3.2 1B Instruct
+        - Modelo: Llama 3.3 70B Instruct
         - Idioma: Português
         - Hospedagem: Hugging Face Spaces
         
         Para melhor desempenho, ajuste os parâmetros de acordo com suas necessidades.
+        Este modelo possui capacidades avançadas de processamento de linguagem natural.
         """)
     
     return demo