Asit03 commited on
Commit
21cf858
·
verified ·
1 Parent(s): 625b615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ import gradio as gr
97
  client = InferenceClient("Asit03/AI_Agent_V2_Merged") # or private repo with token
98
 
99
  def chat(prompt):
100
- response = client.text_generation(prompt, max_new_tokens=150)
101
  return response.strip()
102
 
103
  gr.Interface(fn=chat, inputs="text", outputs="text", title="💬 AI Agent via Inference API").launch()
 
97
  client = InferenceClient("Asit03/AI_Agent_V2_Merged") # or private repo with token
98
 
99
  def chat(prompt):
100
+ response = client.text_generation(prompt, max_new_tokens=1500)
101
  return response.strip()
102
 
103
  gr.Interface(fn=chat, inputs="text", outputs="text", title="💬 AI Agent via Inference API").launch()