Update app.py

app.py

@@ -44,20 +44,14 @@ def generate_response(model, tokenizer, prompt, max_length=512, num_return_seque
     # Tokenizar a resposta gerada
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    # Extrair a parte revelante da resposta
-    response = response.split("### Resposta: ")[-1].strip()
+    # Extrair a parte relevante da resposta
+    if "### Resposta:" in response:
+        response = response.split("### Resposta:")[-1].strip()
 
-    # Encontrar o início da resposta (após "###Resposta:")
-    response_start_index = response.find("### Resposta:") + len("### Resposta:")
-    response_text = response[response_start_index:].strip()
-
-    # A primeira letra da resposta maiuscula
-    response_text = capitalize(response_text)
-
-    # Reconstruir a resposta completa com a primeira letra maiuscula
-    final_response = response[:response_start_index] + " " + response_text
-
-    return final_response
+    # Capitalizar a primeira letra da resposta
+    response = capitalize(response)
+    
+    return response
     
 
 # Define a função para responder às perguntas