Update app.py

app.py

@@ -24,7 +24,7 @@ HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
 def select_model(model_name):
     if model_name in MODEL_LIST:
-        return API_BASE_URL + model_name
+        return f"{API_BASE_URL}{model_name}"
 
 def extend_prompt(input_text):
     if not input_text.strip():
@@ -36,8 +36,9 @@ def extend_prompt(input_text):
     payload = {"inputs": input_text}
 
     try:
-        response = requests.post(API_URL, headers=HEADERS, json=payload).raise_for_status().json()
-        return response[0].get("generated_text", "")
+        response = requests.post(API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
+        return response.json()[0].get("generated_text", "")
     except requests.exceptions.RequestException as e:
         gr.Warning(f"Error in API request: {e}")
         return None