Update app.py

app.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Optional, Tuple
 import gradio as gr
 
 import sambanova_gradio
+from get_fn import get_fn
 
 import modelscope_studio.components.base as ms
 import modelscope_studio.components.legacy as legacy
@@ -155,12 +156,11 @@ with gr.Blocks(css_paths="app.css") as demo:
                 return response_text
 
             # Get the model from sambanova_gradio
-            fn = sambanova_gradio.get_fn(
+            fn = get_fn(
                 model_name='Qwen2.5-Coder-32B-Instruct',
                 preprocess=preprocess,
                 postprocess=postprocess,
-                api_key=YOUR_API_TOKEN,
-                base_url="https://fast-cloud-snova-ai-dev-0-api.cloud.snova.ai/v1"
+                api_key=YOUR_API_TOKEN
             )
 
             response_text = ''