Update app.py

app.py

@@ -19,12 +19,11 @@ def get_publicai_client():
 def _call_llama_space(prompt: str) -> str:
     try:
         client = get_publicai_client()
-        response = client.text_generation(
+        response = client.chat.completions.create(
             model="swiss-ai/apertus-8b-instruct",
-            inputs=prompt,
-            max_new_tokens=512
+            messages=[{"role": "user", "content": prompt}],
         )
-        return response[0]['generated_text'].strip()
+        return response.choices[0].message["content"].strip()
     except Exception as e:
         return f"Error calling PublicAI model: {e}"